onnxruntime/orttraining
Vincent Wang 3654a5d60e
Register Custom Symbolic of torch.einsum for ORTModule (#9590)
* register custom symbolic for einsum

* bugfix for case needs permute at the end

* refactor

* refactor equation parser

* support new case, use ReduceProd

* optimize perf and graph

* remove some Gather node

* add more ut, fix gemm trans fusion
2021-11-18 10:13:58 +08:00
..
orttraining Register Custom Symbolic of torch.einsum for ORTModule (#9590) 2021-11-18 10:13:58 +08:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Switch AMD CI pipeline to use environment image from onnxruntimecibuildenvironment (#9206) 2021-09-28 13:06:16 -07:00