At current stage, TensorRT(up to version 8.6.1) does not support 3D GridSample operator.
This plugin is a custom implementation of the 3D GridSample operator for TensorRT. It is inspired by the GridSample operator from PyTorch, and the code structure is inspired by project onnxparser-trt-plugin-sample.
export PATH=/usr/local/cuda/bin:$PATH
mkdir build && cd build
cmake .. -DTensorRT_ROOT=/usr/local/tensorrt
make
for python code (only on Linux platform), load the plugin with:
import ctypes
success = ctypes.CDLL("build/libgrid_sample_3d_plugin.so", mode = ctypes.RTLD_GLOBAL)
see test_grid_sample3d.py for more details.