Skip to content

Mace
Mace

Xiaomi / Mace

通知 107
Star 40
Fork 27
Mace
Mace
“ca8b7a2760be58670ad9388fddc99a1f4bb0d88a”上不存在“projects/gperftools”
提交 4418e030 编写于 作者: W Wiktor Adamski
浏览文件
操作

Changed code formatting.

上级 ec02c891
隐藏空白更改
内联 并排
Showing with 4 addition and 3 deletion
mace/python/tools/converter_tool/tensorflow_converter.py
浏览文件 @ 4418e030
...@@ -205,8 +205,8 @@ class TensorflowConverter(base_converter.ConverterInterface): ...@@ -205,8 +205,8 @@ class TensorflowConverter(base_converter.ConverterInterface):
} }
pad_type = { pad_type = {
'CONSTANT' : PadType.CONSTANT, 'CONSTANT': PadType.CONSTANT,
'REFLECT' : PadType.REFLECT, 'REFLECT': PadType.REFLECT,
'SYMMETRIC': PadType.SYMMETRIC 'SYMMETRIC': PadType.SYMMETRIC
} }
...@@ -740,7 +740,8 @@ class TensorflowConverter(base_converter.ConverterInterface): ...@@ -740,7 +740,8 @@ class TensorflowConverter(base_converter.ConverterInterface):
if len(tf_op.inputs) == 3: if len(tf_op.inputs) == 3:
constant_value_arg = op.arg.add() constant_value_arg = op.arg.add()
constant_value_arg.name = MaceKeyword.mace_constant_value_str constant_value_arg.name = MaceKeyword.mace_constant_value_str
constant_value = tf_op.inputs[2].eval().astype(np.int32).flat[0] constant_value = tf_op.inputs[2].eval().astype(np.int32) \
.flat[0]
constant_value_arg.i = constant_value constant_value_arg.i = constant_value
self._skip_tensor.add(tf_op.inputs[2].name) self._skip_tensor.add(tf_op.inputs[2].name)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或