Commit Graph

7 Commits (b55c247678e5063598d365bd77b29dec8b62472d)

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