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

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

Description

I have a model include op expand_as/tile, which does things like:
input_x_tensor_shape = (1,256,1,1)
input_y_tensor_shape=(1,256,h,w)
out = input_x.expand_as(input_y) or out = tile(input_x,repeat_times=(1,1,h,w))
out_shape=(1,256,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.

Environment

TensorRT Version:
tensorrt 7.2.3.4
GPU Type:
T4
Nvidia Driver Version:
nvidia440.33
CUDA Version:
10.2
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
2

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

Thanks!

4

It should support