onnx-tensorrt icon indicating copy to clipboard operation
onnx-tensorrt copied to clipboard

Published 20 hours ago • verified publisher icon onnx

Dose trt8.2 has native API to torch.roll?

Open henbucuoshanghai opened this issue 3 years ago • 1 comments

native trt user. tks

henbucuoshanghai avatar Mar 08 '22 09:03 henbucuoshanghai

You can try to use https://github.com/NVIDIA/Torch-TensorRT to see if the operator is supported. Does your model export into ONNX?

kevinch-nv avatar Mar 12 '22 00:03 kevinch-nv