Tensorrt unsupport torch.where()

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Orin
1

I’m trying to convert pytorch—>onnx---->tensorrt, but there are some error such as: In node 6038 (importFallbackPluginImporter): UNSUPPORTED_NODE: Assertion failed: creator && “Plugin not found, are the plugin name, version, and namespace correct?”
And I found it caused by torch.where(). Could you tell me how to fix it? Thanks a lot.

3

I’m closing this topic due to there is no update from you for a period, assuming this issue was resolved.
If still need the support, please open a new topic. Thanks

Hi,

Do you use the trtexec to convert the ONNX model into a TensorRT engine?
If yes, could you share the complete output log with the --verbose flag?

Thanks.