onnxruntime
ba51774a - Add GPU support for DNNL endpoint (#6741)

Commit
4 years ago
Add GPU support for DNNL endpoint (#6741) * Added code for Relugrad with GPU support. Signed-off-by: Chethan Palangotu Keshava <chethan.palangotu.keshava@intel.com> * Add GPU support for DNNL ConvGrad Signed-off-by: George Nash <george.nash@intel.com> * Add GPU support for DNNL MaxPoolGrad Updates to MaxPool for training with GPU Update oneDNN to version 1.8.1 Signed-off-by: George Nash <george.nash@intel.com> * Fixed issues found durring code review - error in code comment - using auto when the direct type would have been better - removed ternary operators that were returning bool values Signed-off-by: George Nash <george.nash@intel.com> Co-authored-by: Chethan Palangotu Keshava <chethan.palangotu.keshava@intel.com>
Author
Parents
Loading