search

lix19937 / tensorrt-insight

Deep insight tensorrt, including but not limited to qat, ptq, plugin, triton_inference, cuda
12 stars 0 forks source link

drive os 6060 trt8510 support grid_sample or not ? #11

Open lix19937 opened 6 months ago

lix19937 commented 6 months ago

@torch.onnx.export
RuntimeError: Exporting the operator grid_sampler to ONNX opset version 11 is not supported
RuntimeError: Exporting the operator grid_sampler to ONNX opset version 13 is not supported.

ops = 11 13 不支持 gird_sample 从16开始支持导出单一节点

op onnx support
GridSample 22, 20, 16

torch.nn.functional.grid_sample

https://onnx.ai/onnx/operators/onnx__GridSample.html

GridSample 22, 20, 16

lix19937 commented 1 month ago

https://docs.nvidia.com/deeplearning/tensorrt/operators/docs/GridSample.html

TensorRT 8.5 supports operators up to Opset 17 from
https://github.com/onnx/onnx-tensorrt/blob/8.5-GA/docs/operators.md

op trt support type
GridSample Y FP32, FP16
LayerNormalization N

torch.nn.functional.grid_sample 的物理含义

https://stackoverflow.com/questions/61570727/how-to-use-pytorchs-grid-sample
https://stackoverflow.com/questions/73300183/understanding-the-torch-nn-functional-grid-sample-op-by-concrete-example
https://blog.csdn.net/weixin_40957452/article/details/121837939