onnx-mlir/test
Aman LaChapelle 24d0a2ac71
Add shape inference and names (#266)
* Add shape inference and names

 - Add shape inference for PRelu
 - Fix shape inference for group conv
   for ConvTranspose
 - Add input and output names for
   graphs (functions)
 - Add support for (u)int8 tensor
   attributes

* Fix format issues

* Revert formatting for gen_onnx_mlir.py

* Pads can have ArrayAttr and DenseElementsAttr so support both

* NumInputs is the number of graph inputs that don't have initializers

* Add test for 2D batchnorm

* Fix typo in define_loops in new 2d BN test

* Change 'name' to 'onnx_node_name'

* Fix Batchnorm for 2D I/O and add lowering test

Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-08-27 15:46:27 -04:00
..
backend Lower SqueezeOp to Krnl dialect (#164) 2020-07-03 16:26:41 +08:00
mlir Add shape inference and names (#266) 2020-08-27 15:46:27 -04: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