提交 44f7d259 编写于 作者: L luxuhui

feat: support FusedBatchNormV2/FusedBatchNormV3 for tensorflow

N/A
Signed-off-by: NLuxuhui <luxuhui@xiaomi.com>
上级 6c7828df
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册