TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard
verified publisher icon NVIDIA

Is NCHW the only supported input layout for IConvolutionLayer in TensorRT?

Open NotUr77 opened this issue 8 months ago • 1 comments

I’m working with TensorRT and need to use IConvolutionLayer. I would like to know if the input tensor is required to be in NCHW format, or if it’s possible to specify a different layout such as NHWC. If custom layouts are supported, how can they be set for the layer? I’m using TensorRT version 10.2. Any clarification would be appreciated!

NotUr77 avatar Apr 25 '25 05:04 NotUr77

You could use IShuffleLayer to permute the axes of the input tensor into NCHW.

yuanyao-nv avatar May 02 '25 20:05 yuanyao-nv