onnxruntime/orttraining
Xueyun Zhu efc8bd738f
add pipeline graph split script (#3275)
* pipeline graph cut

* add element type

* add input wait event and shape info

* shape inference

* support multiple cuts

* format script

* address feedback

* address feedback
2020-03-31 19:30:18 -07:00
..
docker Introduce training changes. 2020-03-11 14:39:03 -07:00
orttraining Don't cast to fp16 in LayernormGrad (#3328) 2020-03-28 19:07:32 -07:00
pytorch_frontend_examples refactor frontend (#3235) 2020-03-19 20:59:41 -07:00
tools add pipeline graph split script (#3275) 2020-03-31 19:30:18 -07:00