colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
onnx-mlir/src/runtime
History
Tung D. Le aea6479ad3
Lower BatchNormalization (test mode) to Krnl dialect (#70) 
* Add ONNXBatchNormalizationTestModeOp and its shape inference

* Lower batchnormalization test mode

* re-use scale, bias, mean, and variance

* Add MLIR tests

* Add e2e tests

* fix typos

* Fix a bug in MLIR tests

* Change type from int to int64_t for indices

* Uncomment e2e tests due to segmentation fault

* Uncomment e2e tests due to segmentation fault

* Revise the code

* [Tian] Fix segmentation fault in e2e tests

* Re-generate onnx.md to include BatchNormalizationTestModeOp

* Reverse an unintentional change

* Fix some typos in comments

* Use convertToMemRefType from the master branch

Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
 		2020-02-20 11:45:40 -05:00
..
CMakeLists.txt flatten src directory structure 2019-12-23 00:13:52 -05:00
data_type.h Enable ONNX Backend Test (#1) 2019-12-22 23:14:57 -05:00
dyn_memref.cpp Lower BatchNormalization (test mode) to Krnl dialect (#70) 2020-02-20 11:45:40 -05:00
dyn_memref.h Enable ONNX Backend Test (#1) 2019-12-22 23:14:57 -05:00
runtime.cpp Enable ONNX Backend Test (#1) 2019-12-22 23:14:57 -05:00
runtime.hpp Enable ONNX Backend Test (#1) 2019-12-22 23:14:57 -05:00