How to add a new layer with Relay,any tutorial? Need:
- Both Python and C++ implement
- for ONNX model
- How to Schedule and optimize
- Both CPU and CUDA
How to add a new layer with Relay,any tutorial? Need:
Same question to me. And my problem is “tvm.error.OpNotImplemented: The following operators are not supported for frontend ONNX: ATen”
Same question to me as well.
This is an ONNX bug, using pytorch1.3 export ONNX which solved problem.