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

about gru support

Open GeorgeBohw opened this issue 4 years ago • 2 comments

hi,i have a question: does onnx-tensorrt support 'gru' operation?

details:now i has a network which has 'gru' operation ,when i use onnx2trt to convert onnx model to trt model,i tell me:

_importer_ctx.tensors().at(output.name()).is_tensor()

so i double that this version can't support 'gru' operation.

thanks!

GeorgeBohw avatar Jan 23 '21 09:01 GeorgeBohw

What TensorRT version are you using? There is an operators.md file in each branch (or docs/operators.md) on master that lists the current onnx-tensorrt supported operators list.

_importer_ctx.tensors().at(output.name()).is_tensor() looks like an error not necessarily related to a particular operator, are you able to share your model?

kevinch-nv avatar Jan 29 '21 23:01 kevinch-nv

@kevinch-nv can you support an sample for RNN convertion? Thanks

leaf918 avatar May 07 '21 10:05 leaf918