Skip to content

Mace
Mace

Xiaomi / Mace

通知 107
Star 40
Fork 27
Mace
Mace
提交 44f7d259 编写于 作者: L luxuhui
浏览文件
操作

feat: support FusedBatchNormV2/FusedBatchNormV3 for tensorflow 

N/A
Signed-off-by: NLuxuhui <luxuhui@xiaomi.com>
上级 6c7828df
隐藏空白更改
内联 并排
Showing with 4 addition and 0 deletion
tools/python/transform/tensorflow_converter.py
浏览文件 @ 44f7d259
...@@ -78,6 +78,8 @@ TFSupportedOps = [ ...@@ -78,6 +78,8 @@ TFSupportedOps = [
'Fill', 'Fill',
'FloorDiv', 'FloorDiv',
'FusedBatchNorm', 'FusedBatchNorm',
'FusedBatchNormV2',
'FusedBatchNormV3',
'Gather', 'Gather',
'GatherV2', 'GatherV2',
'Identity', 'Identity',
...@@ -239,6 +241,8 @@ class TensorflowConverter(base_converter.ConverterInterface): ...@@ -239,6 +241,8 @@ class TensorflowConverter(base_converter.ConverterInterface):
TFOpType.Fill.name: self.convert_fill, TFOpType.Fill.name: self.convert_fill,
TFOpType.FloorDiv.name: self.convert_elementwise, TFOpType.FloorDiv.name: self.convert_elementwise,
TFOpType.FusedBatchNorm.name: self.convert_fused_batchnorm, 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.Gather.name: self.convert_gather,
TFOpType.GatherV2.name: self.convert_gather, TFOpType.GatherV2.name: self.convert_gather,
TFOpType.Identity.name: self.convert_identity, TFOpType.Identity.name: self.convert_identity,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或