onnx-mlir/test/mlir/onnx
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
..
onnx_bundle_memory_pool.mlir Assorted fixes for memory pool related passes (#244) 2020-08-11 17:34:59 -04:00
onnx_canonicalization.mlir Fuse convolution and batch normalization (#253) 2020-08-18 16:41:40 +08:00
onnx_constprop.mlir Fuse convolution and batch normalization (#253) 2020-08-18 16:41:40 +08:00
onnx_decompose.mlir ScalerOp support non-float input using CastOp (#231) 2020-07-24 10:57:52 -04:00
onnx_elide_constants.mlir Rework output to improve readability of intermediate MLIR code. (#87) 2020-04-24 16:15:36 -04:00
onnx_enable_memory_pool.mlir [NFC] Rename passes for stylistic consistency. (#232) 2020-07-31 21:37:35 +08:00
onnx_krnl_global_elision.mlir Remove optimize_loops/return_loops op. (#200) 2020-07-08 12:49:15 +08:00
onnx_lowering.mlir Add shape inference and names (#266) 2020-08-27 15:46:27 -04:00
onnx_lowering_with_dealloc.mlir [NFC] Rename passes for stylistic consistency. (#232) 2020-07-31 21:37:35 +08:00
onnx_shape_inference.mlir Add shape inference and names (#266) 2020-08-27 15:46:27 -04:00
onnx_shape_inference_maxpool.mlir Add the split marker to test files (#90) 2020-04-16 15:17:27 +08:00
onnx_structure.mlir Seq type (#199) 2020-07-31 20:05:59 +08:00