You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
the model i used has some opration written by c++, using torch.utils.cpp_extention with a .cpp file. I want to convert the model into trt model but failed with the error NVALID_ARGUMENT: Cannot find binding of given name: input_0.
how can i write a converter for the operation written by c++?
The text was updated successfully, but these errors were encountered:
First, you need to implement the custom ops you need as a TensorRT plugin, here are some examples.
After you have your plugin, add a creator in torch2trt_dynamic/plugins, note that if you did not implement your plugin inside amirstan_plugin, you need to load library ctypes.CDLL('xxxxx.so')
the model i used has some opration written by c++, using torch.utils.cpp_extention with a .cpp file. I want to convert the model into trt model but failed with the error NVALID_ARGUMENT: Cannot find binding of given name: input_0.
how can i write a converter for the operation written by c++?
The text was updated successfully, but these errors were encountered: