onnx-mlir/test/mlir
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
..
conversion [NFC] Rename passes for stylistic consistency. (#232) 2020-07-31 21:37:35 +08:00
krnl [NFC] Rename passes for stylistic consistency. (#232) 2020-07-31 21:37:35 +08:00
onnx Add shape inference for Ops used by BERT (#249) 2020-08-07 13:08:00 +08:00
transform Shape inference for some ops in RNN models (#207) 2020-07-22 10:15:56 -04:00
CMakeLists.txt Compiling Models with Large Constant Arrays (#146) 2020-06-12 10:27:05 +08:00
lit.cfg.py Compiling Models with Large Constant Arrays (#146) 2020-06-12 10:27:05 +08:00
lit.site.cfg.py.in Remove whole archive linkage (#173) 2020-06-19 00:21:27 +08:00