colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
onnx-mlir/utils
History
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
..
CMakeLists.txt Add shape inference for several ops (#163) 2020-06-09 14:55:49 +08:00
build-mlir.cmd Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
build-mlir.sh Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
clone-mlir.sh Move to more recent LLVM commit ID (#131) 2020-05-20 15:45:42 +08:00
debug.py Minor bugfix. (#166) 2020-06-08 14:03:35 -04:00
gen_onnx_mlir.py Add shape inference for several ops (#163) 2020-06-09 14:55:49 +08:00
install-mlir.cmd Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
install-mlir.sh Add support for Windows using Visual Studio Compiler (#86) 2020-04-19 22:11:24 +08:00
install-onnx-mlir.cmd Enable Build on Windows again and Update CI badge (#115) 2020-05-12 23:00:19 +08:00
install-onnx-mlir.sh Rapid check test (#141) 2020-06-08 10:18:55 +08:00
setup-onnx-mlir-windows.cmd Enable Build on Windows again and Update CI badge (#115) 2020-05-12 23:00:19 +08:00