NVIDIA Developer Forums

Upsampling NCHW format for TensorRT plugin

AI & Data Science Deep Learning (Training & Inference) TensorRT

edit_or August 30, 2019, 9:51am 1

I am implementing TensorRT plugin and like to make sure my upsampling is correct format for TensorRT.

I have detail explanation in the following link.

[url]tensorflow - Upsampling Tensor for TensorRT - Stack Overflow

Is Method_1 or Method_2 correct way of upsampling for NCHW format?

Topic		Replies	Views	Activity
PluginFormat has to be kNCHW? TensorRT	0	571	August 29, 2019
Should I retrain the model when the TensorRT uses NCHW as input and the pre-trained model uses NHWC format TensorRT	1	540	August 27, 2019
TensorRT plugin input data format TensorRT	0	371	September 2, 2019
TensorRT memory layout TensorRT	1	1222	November 25, 2019
NCHW and atrous in tensorflow Jetson TX2	2	618	October 18, 2021
TensorRT UFF Tensorflow NHWC (channels last) to NCHW (channels first) conversion buggy TensorRT	2	3212	November 2, 2019
TensorRT specify layer NCHW -> NHWC Jetson TX2	7	2020	October 18, 2021
TensorRT support NHWC model? Jetson TX2	4	3195	October 18, 2021
Does TensorRT 3.0 C++ API only works for NCHW? Jetson TX2	5	1543	October 18, 2021
TensorRT: .wts format for tensorflow model weights in sampleCharRNN GPU-Accelerated Libraries	0	1058	August 22, 2017