onnx-mlir/test/mlir/onnx
Aman LaChapelle ca185002f2
Add shape inference for several ops (#163)
* 1. Add shape inference for the following ops:
 - Atan
 - Tan
 - Sin
 - Cast
 - ConvTranspose
 - Flatten
 - DynamicQuantizeLinear
 - QuantizeLinear
 - DequantizeLinear
 - ConvInteger

2. Import attributes for generic nodes
3. Fixes for cases where .cast<> should be .isa<> (ONNXConcat::inferShapes)

* Fix foormatting issues

* Address comments:
 - SmallVector<> * -> SmallVectorImpl<> &
 - switch-case -> helper function
   - Inside helper function, preserve signed-ness
 - add TODOs

* Can't use signed integers yet in convertONNXTypeToMLIRType, add TODO

Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-06-09 14:55:49 +08:00
..
onnx_canonicalization.mlir Implement PadOp based on attribute promotion (#71) 2020-05-15 13:19:28 +08:00
onnx_constprop.mlir Constprop (#162) 2020-06-08 15:45:32 -04:00
onnx_decompose.mlir Add the split marker to test files (#90) 2020-04-16 15:17:27 +08:00
onnx_elide_constants.mlir Rework output to improve readability of intermediate MLIR code. (#87) 2020-04-24 16:15:36 -04:00
onnx_krnl_global_elision.mlir Rework output to improve readability of intermediate MLIR code. (#87) 2020-04-24 16:15:36 -04:00
onnx_lowering.mlir Add AffineScope trait to KrnlIterateOp and enable affine-loop-fusion pass (#140) 2020-06-08 15:36:27 +08:00
onnx_lowering_with_dealloc.mlir Add AffineScope trait to KrnlIterateOp and enable affine-loop-fusion pass (#140) 2020-06-08 15:36:27 +08:00
onnx_shape_inference.mlir Add shape inference for several ops (#163) 2020-06-09 14:55:49 +08:00
onnx_shape_inference_maxpool.mlir Add the split marker to test files (#90) 2020-04-16 15:17:27 +08:00