提交 137c106e 编写于 作者: 叶剑武

Merge branch 'master' into 'master'

init shape before deciding if append dims

See merge request !292
...@@ -786,7 +786,6 @@ class TFConverter(object): ...@@ -786,7 +786,6 @@ class TFConverter(object):
reshape_op = self.tf_graph[softmax_op.name][0] reshape_op = self.tf_graph[softmax_op.name][0]
self.unused_tensor.add(get_input_tensor(reshape_op, 1).name) self.unused_tensor.add(get_input_tensor(reshape_op, 1).name)
if reshape_op.outputs[0].shape.ndims == 2:
shape = [dim.value for dim in reshape_op.outputs[0].shape] shape = [dim.value for dim in reshape_op.outputs[0].shape]
if len(shape) == 2: if len(shape) == 2:
shape = [1, 1, shape[0], shape[1]] shape = [1, 1, shape[0], shape[1]]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册