From bc7bd0a2abc236e86109c528abe8731f40cdc353 Mon Sep 17 00:00:00 2001 From: Liangliang He Date: Thu, 30 Nov 2017 14:16:30 +0800 Subject: [PATCH] Update tf_ops_stats.py to print not fully defined shape --- mace/python/tools/tf_ops_stats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/python/tools/tf_ops_stats.py b/mace/python/tools/tf_ops_stats.py index 9301b3f1..d1016aff 100644 --- a/mace/python/tools/tf_ops_stats.py +++ b/mace/python/tools/tf_ops_stats.py @@ -68,7 +68,7 @@ def main(unused_args): if input_name.endswith('weights:0') and input_name in tensor_shapes: ksize = tensor_shapes[input_name] 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) hist_inc(stats, key) elif op.type in ['FusedResizeAndPadConv2D']: -- GitLab