提交 b742f824 编写于 作者: 李超

Merge branch 'fused_batch_norm_v3' into 'master'

feat: support FusedBatchNormV2/FusedBatchNormV3 for tensorflow

See merge request applied-machine-learning/sysml/mace!1296
......@@ -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.
先完成此消息的编辑!
想要评论请 注册