onnxruntime/orttraining/tools
Xueyun Zhu f1ba9aaf34
Add pipeline transformer for wait/record node (#3513)
* pipeline transformer

* clean up

* address feedback

* add record/wait for first stage and updated split script

* address feedback

* make recv/send signal as initializer

* merge

* address feedback

* unify input and initializer

* address feedback and bug fix

* minor fix

* windows build

* fix
2020-04-22 23:28:01 -07:00
..
ci_test MaxBatchSize E2E Test (#3454) 2020-04-15 09:50:44 +08:00
mnist_model_builder Introduce training changes. 2020-03-11 14:39:03 -07:00
scripts Add pipeline transformer for wait/record node (#3513) 2020-04-22 23:28:01 -07:00