onnx-mlir

Commit Graph

Author	SHA1	Message	Date
TUNG LEDUC	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
TUNG LEDUC	45608282e0	[MLIR] Add support for Relu (#392 ) * Add support for Relu * Add comments	2019-12-21 01:38:16 -05:00
TUNG LEDUC	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
TUNG LEDUC	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
GHEORGHE-TEOD BERCEA	652ce4b7d4	Add test for checking lowering of Add op to KRNL IR (#385 ) * Add test for checking lowering of Add op to KRNL IR. * Add test file.	2019-12-21 01:20:36 -05:00

5 Commits