diff --git a/tools/python/transform/tensorflow_converter.py b/tools/python/transform/tensorflow_converter.py index 29d20b7f495685ab18574acc4f34d9fededc1a7a..cfd24ace00945869a7e5a935257cd1c926916a68 100644 --- a/tools/python/transform/tensorflow_converter.py +++ b/tools/python/transform/tensorflow_converter.py @@ -78,6 +78,8 @@ TFSupportedOps = [ 'Fill', 'FloorDiv', 'FusedBatchNorm', + 'FusedBatchNormV2', + 'FusedBatchNormV3', 'Gather', 'GatherV2', 'Identity', @@ -239,6 +241,8 @@ class TensorflowConverter(base_converter.ConverterInterface): TFOpType.Fill.name: self.convert_fill, TFOpType.FloorDiv.name: self.convert_elementwise, TFOpType.FusedBatchNorm.name: self.convert_fused_batchnorm, + TFOpType.FusedBatchNormV2.name: self.convert_fused_batchnorm, + TFOpType.FusedBatchNormV3.name: self.convert_fused_batchnorm, TFOpType.Gather.name: self.convert_gather, TFOpType.GatherV2.name: self.convert_gather, TFOpType.Identity.name: self.convert_identity,