onnxruntime/onnxruntime
Oliver Rausch 9c3153acd6
Improve shape inference for OneHot (#4452)
* Improve shape inference for OneHot

Attempt to get the depth parameter before adding a new symbolic dimension.

* Update symbolic shape infer

* Nit
2020-08-07 14:05:20 -07:00
..
contrib_ops Add the framework to support prepack (#4413) 2020-08-07 09:39:19 -07:00
core Improve shape inference for OneHot (#4452) 2020-08-07 14:05:20 -07:00
featurizers_ops/cpu
gsl
python Update notebook for mac since onnxruntime 1.3 or 1.4 in mac does not have openmp (#4732) 2020-08-07 14:01:48 -07:00
test Add the framework to support prepack (#4413) 2020-08-07 09:39:19 -07:00
tool/etw
.style.yapf Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
__init__.py Bump version to 1.4.0 (#4496) 2020-07-13 17:09:18 -07:00
ReformatSource.ps1
ReformatSourcePython.bat Update GPT2 Model Benchmark Script to Support IO Binding (#4088) 2020-06-01 15:07:48 -07:00
VSCodeCoverage.runsettings