I compile and tune a pytorch detection model using tvm, but the detection layer is not supported when converting to onnx model format, so I remove it but only keep the box prediction and classification prediction, so how to do detection and nms in tvm api to avoid convert to numpy and run the post process step in cpu? Thanks.
