From 44f7d25917c6cfa8e04dc84c4f29af3163388107 Mon Sep 17 00:00:00 2001 From: luxuhui Date: Mon, 21 Sep 2020 17:42:38 +0800 Subject: [PATCH] feat: support FusedBatchNormV2/FusedBatchNormV3 for tensorflow N/A Signed-off-by: Luxuhui --- tools/python/transform/tensorflow_converter.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/python/transform/tensorflow_converter.py b/tools/python/transform/tensorflow_converter.py index 29d20b7f..cfd24ace 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, -- GitLab