7e3f96e642
[MLIR] Add support for Reciprocal ( #397 )
...
* Added support for Reciprocal
* Fixed format
2019-12-21 02:07:44 -05:00
5ed79083d5
[MLIR] Add support for Max, Min, Sum, Elu, Selu, LeakyRelu, HardSigmoid ( #395 )
...
* Lower ONNXSumOp
* Add inferShapes() and test cases
* Load the first operand to the result
* Update SharingWork.md
* Update SharingWork.md
* Update SharingWork.md
* Add support for Max, Min
* Pass operation instead of location to mapToLowerScalarOp
* Add support for Elu, Selu, LeakyRelu, HardSigmoid
* Add test cases
* Update SharingWork.md
* Rewrite the part of lowering variadic ops and use it for binary ops
* Use two diffenrent templates for Unary and Variadic Ops
* Revise the code
2019-12-21 02:02:09 -05:00
45608282e0
[MLIR] Add support for Relu ( #392 )
...
* Add support for Relu
* Add comments
2019-12-21 01:38:16 -05:00
1c3176bf9f
[MLIR] Lower ONNX element-wise unary ops: Exp, Tanh, Sinh, Cosh, Sigmoid ( #389 )
...
* Lower ExpOp
* Lower ONNXTanhOp
* Lower Exp Tanh, Sinh, and Cosh
* Lower ONNX Sigmoid op
* Merge
* Specialize template lowerScalarOp
* Unify ONNXEWUnaryOpLowering and ONNXEWBinaryOpLowering
* Support multiple types
* Reformat the code
* Add test cases
* Reformat the code
* Change names
* Apply clang-format
* Update variable names
2019-12-21 01:37:29 -05:00
0048f2fd86
clean up
2019-12-21 01:36:03 -05:00
c3ef1d93ae
[MLIR] Lower ONNX element-wise binary ops: Mul, Div, Sub, And, Or, Xor ( #388 )
...
* Lower ONNX element-wise binary ops: Mul, Div, Sub, And, Or, Xor
* Edit gen_doc.py to avoid changes about AnyTypeOf<[AnyMemRef, AnyTensor]>
* Miss a space
* Add tests
* Shorten ONNXElementWiseBinaryOpLowering into ONNXEWBinaryOpLowering
* Move lowering patterns into runOnModule()
* Redundant space
2019-12-21 01:35:31 -05:00
7fb2f80dce
[MLIR] Add support for dealloc insertion ( #386 )
...
* Add support for dealloc op.
* Check dealloc for returned result not present.
2019-12-21 01:34:48 -05:00
Haruki Imai
TUNG LEDUC
Tian Jin
GHEORGHE-TEOD BERCEA