From 40f91c18563f5d41e1825be139f4614866dd22af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= Date: Tue, 12 Dec 2017 14:59:40 +0800 Subject: [PATCH] Add max_byte_size for flatten --- mace/python/tools/tf_dsp_converter_lib.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mace/python/tools/tf_dsp_converter_lib.py b/mace/python/tools/tf_dsp_converter_lib.py index 209173e9..e9eae636 100644 --- a/mace/python/tools/tf_dsp_converter_lib.py +++ b/mace/python/tools/tf_dsp_converter_lib.py @@ -149,6 +149,7 @@ def convert_ops(unresolved_ops, resolved_ops, net_def, output_node, dsp_ops): elif is_node_flatten_reshape(first_op): op_def.type = 'Flatten' op_def.input.extend([t.name for t in first_op.inputs]) + op_def.out_max_byte_size.extend([max_elem_size(out) for out in first_op.outputs]) convert_op_outputs(op_def, first_op) elif dsp_ops.has_op(first_op.type): op_def.input.extend([t.name for t in first_op.inputs]) -- GitLab