24d0a2ac71
* 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> |
||
---|---|---|
.. | ||
onnx_bundle_memory_pool.mlir | ||
onnx_canonicalization.mlir | ||
onnx_constprop.mlir | ||
onnx_decompose.mlir | ||
onnx_elide_constants.mlir | ||
onnx_enable_memory_pool.mlir | ||
onnx_krnl_global_elision.mlir | ||
onnx_lowering.mlir | ||
onnx_lowering_with_dealloc.mlir | ||
onnx_shape_inference.mlir | ||
onnx_shape_inference_maxpool.mlir | ||
onnx_structure.mlir |