TensorRT4.0.1.6 BatchNormalization support problem


Does tf.layers.batch_normalization() support by the operation “FusedBatchNorm”?

When I used this function in my model and convert it to .uff file, I got some warnings: No conversion function registered for the layer: Merge (and Switch) yet.

Is the tf.layers.batch_normalization() function including these three operation “FusedBatchNorm”, “Merge”, “Switch”, but only “FusedBatchNorm” supported by TensorRT?



TRT4 UFF Parser does not support Merge or Switch operation. The list of supported TensorFlow ops are mentioned on the dev guide, https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html#support_op

Recommend the option to implement a plugin layer for the unsupported operation or modify the original network to remove this operation. Could try using graphsurgeon for this.