saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-07 17:15:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining
History
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
..
core support build option to include external graph transformers (#9478) 2021-11-15 08:16:20 -08:00
eager Change TensorShape to typically not allocate heap memory (#9542) 2021-11-08 10:29:54 -08:00
models Change TensorShape to typically not allocate heap memory (#9542) 2021-11-08 10:29:54 -08:00
python Register Custom Symbolic of torch.einsum for ORTModule (#9590) 2021-11-18 10:13:58 +08:00
test Register Custom Symbolic of torch.einsum for ORTModule (#9590) 2021-11-18 10:13:58 +08:00
training_ops Address some code scan issues. (#9752) 2021-11-16 10:24:46 -08:00