onnxruntime/tools
Vincent Wang dac24f7d63
Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590)
* build with libtorch and impl torchembedding

* fix op shape infer

* local commit

* atenfunctionop

* call aten operator from online extension

* rollback build.py

* resolve comments

* bugfix

* fix build

* fix ortmodule test

* remove external outputs, resolve comments

* resolve comments

* export embedding to microsoft::atenop

* bugfix
2021-05-13 09:24:27 +08:00
..
ci_build Add minsdkver for AAR and AndroidTest (#7669) 2021-05-12 16:01:25 -07:00
doc Add auto doc gen for ORTModule API during CI build (#7046) 2021-03-22 10:20:33 -07:00
nuget Enable Microsoft.Ai.MachineLearning package to work on .NET5 down to 17763 Windows SDK (#7522) 2021-05-01 00:56:36 -07:00
perf_util
python Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590) 2021-05-13 09:24:27 +08:00
test