how to convert a tensorflow graph containing indexing operation and expand_dims to UFF

Autonomous Machines Jetson & Embedded Systems Jetson TX2
1

It seems that tensorrt does not support expand_dims and indexing operation of tensorflow. Is plugin the only way to convert the tensorflow model to UFF?

2

Hi,

You can find the TensorRT supported layer on our User Guide:
/usr/share/doc/tensorrt/TensorRT-3-User-Guide.pdf.gz

For non-supported layer, please implement it with Plugin API.
Thanks.