提交 66e18bfe 编写于 作者: 刘琦

Merge branch 'feature_wuch' into 'master'

fix dsp converter

See merge request !585
...@@ -480,6 +480,7 @@ def add_input_output_info(net_def, input_node, output_node, graph, dtype): ...@@ -480,6 +480,7 @@ def add_input_output_info(net_def, input_node, output_node, graph, dtype):
get_tensor_name_from_op(output_node, 0)) get_tensor_name_from_op(output_node, 0))
input_info = net_def.input_info.add() input_info = net_def.input_info.add()
input_info.name = input_node
input_info.dims.extend(input_tensor.shape.as_list()) input_info.dims.extend(input_tensor.shape.as_list())
input_info.data_type = dtype input_info.data_type = dtype
if dtype == mace_pb2.DT_UINT8: if dtype == mace_pb2.DT_UINT8:
...@@ -489,6 +490,7 @@ def add_input_output_info(net_def, input_node, output_node, graph, dtype): ...@@ -489,6 +490,7 @@ def add_input_output_info(net_def, input_node, output_node, graph, dtype):
input_info.data_type = mace_pb2.DT_FLOAT input_info.data_type = mace_pb2.DT_FLOAT
output_info = net_def.output_info.add() output_info = net_def.output_info.add()
output_info.name = output_node
output_info.dims.extend(output_tensor.shape.as_list()) output_info.dims.extend(output_tensor.shape.as_list())
output_info.data_type = dtype output_info.data_type = dtype
if dtype == mace_pb2.DT_UINT8: if dtype == mace_pb2.DT_UINT8:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册