Description
Converting graph using tftrt with batch>1 will not convert “AddN”
Environment
TensorRT Version:
7.1
GPU Type:
Jetson Xavier
Nvidia Driver Version:
Jetpack 4.2.4
CUDA Version:
10.2
CUDNN Version:
8.0.0
Operating System + Version:
Ubuntu 18.04
Python Version (if applicable):
Python 3.6
TensorFlow Version (if applicable):
tensorflow-2.1.0+nv20.4-cp36-cp36m-linux_aarch64.whl
Baremetal or Container (if container which image + tag):
Baremetal
Steps To Reproduce
Convert graph with batch 1: AddN is inside the engine:
tensorflow/compiler/tf2tensorrt/segment/[segment.cc:460](http://segment.cc:460/)] There are 32 ops of 16 different types in the graph that are not converted to TensorRT: Identity, GatherV2, Reshape, Size, Pack, NonMaxSuppressionV5, Tile, AddV2, ArgMax, Placeholder, NoOp, Cast, StridedSlice, ExpandDims, Mul, Squeeze,
Convert graph with batch >1: AddN is not converted and engine is split:
tensorflow/compiler/tf2tensorrt/segment/[segment.cc:460](http://segment.cc:460/)] There are 61 ops of 17 different types in the graph that are not converted to TensorRT: ConcatV2, Squeeze, Mul, ExpandDims, StridedSlice, Cast, Placeholder, Pack, NonMaxSuppressionV5, AddN, Tile, AddV2, ArgMax, NoOp, GatherV2, Size, Reshape,
What can cause this?