onnx-mlir/test/mlir
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
..
conversion Bug fix, ensure krnl.iterate can lower in the degenerate case. (#78) 2020-04-10 23:27:00 +08:00
krnl Add the split marker to test files (#90) 2020-04-16 15:17:27 +08:00
onnx Add shape inference for several ops (#163) 2020-06-09 14:55:49 +08:00
transform Implement PadOp based on attribute promotion (#71) 2020-05-15 13:19:28 +08:00
CMakeLists.txt Name change for tests, to be check-onnx-(lit | backend) (#62) 2020-03-31 10:06:14 -04:00
lit.cfg.py [NFC] Change ONNF based names to ONNX-MLIR (#32) 2020-03-17 09:16:33 -04:00
lit.site.cfg.py.in Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00