colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
onnx-mlir/utils
History
albertomagni-ms e1386b0689
Add shape inference for Ops used by BERT (#249) 
* Add shape inference for Ops used by BERT

* Erf
* Pow
* ReduceMean
* Dropout
* Expand
  https://github.com/onnx/onnx/blob/master/docs/Operators.md#expand
  Deduce the value of the shape operand by looking at the producer
  of the operand.
  Currently supported producers are: onnx.Constant and onnx.Shape.

* Add corresponding tests for each op.

* Sort the list of ops with shape inference in gen_onnx_mlir.py
in alphabetic order for clarity.

* Restart CI

Co-authored-by: Tian Jin <tjingrant@gmail.com>
 		2020-08-07 13:08:00 +08:00
..
CMakeLists.txt Merge onnx ml into onnx (#176) 2020-06-22 20:01:56 -04:00
build-mlir.cmd Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
build-mlir.sh Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
clone-mlir.sh Update llvm version (#187) 2020-07-07 21:26:00 +08:00
debug.py Minor bugfix. (#166) 2020-06-08 14:03:35 -04:00
gen_onnx_mlir.py Add shape inference for Ops used by BERT (#249) 2020-08-07 13:08:00 +08:00
install-mlir.cmd Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
install-mlir.sh Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
install-onnx-mlir.cmd Enable Build on Windows again and Update CI badge (#115) 2020-05-12 23:00:19 +08:00
install-onnx-mlir.sh Rapid check test (#141) 2020-06-08 10:18:55 +08:00
setup-onnx-mlir-windows.cmd Catch build failure on Windows-CI (Azure Pipelines) (#241) 2020-08-06 16:53:59 +08:00