onnxruntime/tools
Vincent Wang c8d210de29
Decouple Forward and Backward of ATenOp (#8301)
* atenop for inference

* assert if dtype mismatch

* atenop config in frontend

* fix orttrainer test

* gradient def not only for ATenOp

* bugfix

* fix gradient input shape and type issue

* fix after merge master
2021-07-23 16:53:26 +08:00
..
ci_build Implement SparseTensor and infrastructure suppport and advance ONNX commit (#8038) 2021-07-22 15:24:36 -07:00
doc Add graphviz into Dockerfile images for Python API documentation (#7819) 2021-06-02 16:12:54 -07:00
nuget Move one function from cuda_provider_factory.h (#8407) 2021-07-19 17:55:59 -07:00
perf_util
python Decouple Forward and Backward of ATenOp (#8301) 2021-07-23 16:53:26 +08:00
test