onnx-mlir/test
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
..
backend Lower SqueezeOp to Krnl dialect (#164) 2020-07-03 16:26:41 +08:00
mlir Add shape inference for Ops used by BERT (#249) 2020-08-07 13:08:00 +08:00
numerical Add emitjni target (#204) 2020-07-11 13:23:13 +08:00
CMakeLists.txt Rapid check test (#141) 2020-06-08 10:18:55 +08:00