I have Tensorflow model converting to TensorRT and upsample op needs to make a plugin.
CUDA code with a plugin is made as follows.
For example 1x3x3x2 (NHWC) tensor is upscaled to 1x12x12x2 (NHWC) tensor as follow.
My CUDA code is expecting data in the format
But output from CUDA looks like data is fed to plugin as
How to solve the issue?