e1386b0689
* 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> |
||
---|---|---|
.. | ||
conversion | ||
krnl | ||
onnx | ||
transform | ||
CMakeLists.txt | ||
lit.cfg.py | ||
lit.site.cfg.py.in |