2019-12-08 05:53:43 +08:00
|
|
|
# ONNX to KRNL work
|
|
|
|
|
|
|
|
Please add row in first table when adding a new ONNX operator, and move to second table when work is completed
|
|
|
|
|
|
|
|
ONNX operations for which some work is needed.
|
|
|
|
* M for multi-broadcast, U for unidir-broadcast
|
|
|
|
|
|
|
|
|
[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-12 10:49:50 +08:00
|
|
|
| ONNX Oper | Person working on it | ONNX 2 KRNL | Basic functionality | Extended functionality (e.g. broadcast) |
|
|
|
|
| ---------- | --------------------- | -------------- | --------------------- | ---------------------------------------- |
|
2019-12-20 00:28:06 +08:00
|
|
|
| Add | Tung (updated) | v | v | M |
|
|
|
|
| And | Tung | v | v | M |
|
|
|
|
| Cosh | Tung | v | v | |
|
|
|
|
| Div | Tung | v | v | M |
|
[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-12 10:49:50 +08:00
|
|
|
| Elu | Tung | v | v | |
|
|
|
|
| Exp | Tung | v | v | |
|
|
|
|
| FullGemm | | | | noU |
|
|
|
|
| Gemm | | | | noU |
|
|
|
|
| HardSigmoid | Tung | v | v | |
|
|
|
|
| LeakyRelu | Tung | v | v | |
|
|
|
|
| MatMul | | | | noM |
|
2019-12-20 00:28:06 +08:00
|
|
|
| Max | Tung | v | v | M |
|
|
|
|
| Min | Tung | v | v | M |
|
|
|
|
| Mul | Tung | v | v | M |
|
|
|
|
| Or | Tung | v | v | M |
|
[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-12 10:49:50 +08:00
|
|
|
| Relu | Tung | v | v | |
|
|
|
|
| Selu | Tung | v | v | |
|
|
|
|
| Sigmoid | Tung | v | v | |
|
|
|
|
| Sinh | Tung | v | v | |
|
2019-12-20 00:28:06 +08:00
|
|
|
| Sub | Tung | v | v | M |
|
|
|
|
| Sum | Tung | v | v | M |
|
[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-12 10:49:50 +08:00
|
|
|
| Tanh | Tung | v | v | |
|
2019-12-20 00:28:06 +08:00
|
|
|
| Xor | Tung | v | v | M |
|
2019-12-08 05:53:43 +08:00
|
|
|
|
|
|
|
|
|
|
|
ONNX operations for which the work is completed (full functionality) and tested
|
|
|
|
|
[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-12 10:49:50 +08:00
|
|
|
| ONNX Oper | Person working on it | Initial work | Basic functionality | Extended functionality (e.g. broadcast) |
|
|
|
|
| ---------- | ---------------------- | -------------- | --------------------- | ---------------------------------------- |
|