onnxruntime/orttraining
ytaous 0008e92b4e
Internal ReduceSum op that accepts axes as input (#4522)
* Initial change, to add ReduceSumTraining cpu op

* cpu support

* cuda support + more UTs

* on comments + UT

* no op support for {} axes with new attr - noop_with_empty_axes

* on comments

* fix build

* on comments

Co-authored-by: aishwarya bhandare <aibhanda@microsoft.com>
Co-authored-by: Ethan Tao <ettao@microsoft.com>
2020-07-20 21:05:00 -07:00
..
orttraining Internal ReduceSum op that accepts axes as input (#4522) 2020-07-20 21:05:00 -07:00
pytorch_frontend_examples Remove ORTModel from frontend API (#3825) 2020-05-11 18:20:33 -07:00
tools Update convergence baseline for ci_test. (#4465) 2020-07-09 15:29:36 +08:00