Skip to content

Mace
Mace

慢慢CG / Mace
与 Fork 源项目一致

Fork自 Xiaomi / Mace

通知 1
Star 0
Fork 0
Mace
Mace
提交 bc7bd0a2 编写于 作者: L Liangliang He
浏览文件
操作

Update tf_ops_stats.py to print not fully defined shape

上级 2e062c5a
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mace/python/tools/tf_ops_stats.py
浏览文件 @ bc7bd0a2
...@@ -68,7 +68,7 @@ def main(unused_args): ...@@ -68,7 +68,7 @@ def main(unused_args):
if input_name.endswith('weights:0') and input_name in tensor_shapes: if input_name.endswith('weights:0') and input_name in tensor_shapes:
ksize = tensor_shapes[input_name] ksize = tensor_shapes[input_name]
break break
print('%s(padding=%s, strides=%s, ksize=%s, format=%s) %s => %s' % (op.type, padding, strides, ksize, data_format, op.inputs[0].shape.as_list(), op.outputs[0].shape.as_list())) print('%s(padding=%s, strides=%s, ksize=%s, format=%s) %s => %s' % (op.type, padding, strides, ksize, data_format, op.inputs[0].shape, op.outputs[0].shape))
key = '%s(padding=%s, strides=%s, ksize=%s, format=%s)' % (op.type, padding, strides, ksize, data_format) key = '%s(padding=%s, strides=%s, ksize=%s, format=%s)' % (op.type, padding, strides, ksize, data_format)
hist_inc(stats, key) hist_inc(stats, key)
elif op.type in ['FusedResizeAndPadConv2D']: elif op.type in ['FusedResizeAndPadConv2D']:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或