Does tensorrt suport op like "expand_as" or "tile" for dynamic shape input?


I have a model include op expand_as/tile, which does things like:
input_x_tensor_shape = (1,256,1,1)
out = input_x.expand_as(input_y) or out = tile(input_x,repeat_times=(1,1,h,w))

which dynamic shape means h,w is not fixed while inference,
Does tensorrt supoort this situation?

A clear and concise description of the bug or issue.


TensorRT Version:
GPU Type:
Nvidia Driver Version:
CUDA Version:
CUDNN Version:
cudnn 8.1.1
Operating System + Version:
centos 7.9
Python Version (if applicable):

TensorFlow Version (if applicable):
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):

Relevant Files

Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro
  • Exact steps/commands to run your repro
  • Full traceback of errors encountered

Hi ,
We recommend you to check the supported features from the below link.
You can refer below link for all the supported operators list.
For unsupported operators, you need to create a custom plugin to support the operation


It should support