Commit Graph

7 Commits (f76fee644cf045efc3a9b7729e1042cfbe688fe0)

Author SHA1 Message Date
fengjiayi e7d8e16a66 update softmax_mkldnn_op
7 years ago
Jacek Czaja 98f3ad3ba1 - MKLDNN Softmax Grad Op
8 years ago
Jacek Czaja 7bf00c3a93 - First draft of reusing of softmax mkldnn primitives
8 years ago
Siddharth Goyal 5fe1fe3a27
Fix signed/unsigned comparison warning (#10211)
8 years ago
Jacek Czaja acdf7cbd19 - Added EPS for softmax MKLDNN op
8 years ago
Abhinav Arora 7b86da7195 Fix CPPLint errors in operators (#9826)
8 years ago
Jacek Czaja 3b95b55f07 - Softmax MKLDNN primitive integration
8 years ago