tensorrt does not support tf.expand_dims: Unsupported operation _ExpandDims

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

I want to accelerate inference with tensorrt for tensorflow model. First i convert the pb model to uff model, Then i parse uff model, but got the following error:

ERROR: UFFParser: Validator error: ExpandDims_36: Unsupported operation _ExpandDims
ERROR: sample_uff_mnist: Fail to parse

It seems tensorrt not support tf.expand_dims operation. What should i do to solve this problem?

2

Hi,

  1. What version of TensorRT are you using?
  2. Can you provide a small package that can reproduce this problem with the model and conversion code?

Thanks,
NVIDIA Enterprise Support