未验证 提交 0c37fdb5 编写于 作者: J Jason 提交者: GitHub

Update tf_op_mapper.py

上级 b95d36f0
...@@ -718,7 +718,7 @@ class TFOpMapper(OpMapper): ...@@ -718,7 +718,7 @@ class TFOpMapper(OpMapper):
if input.tf_data_format == "NHWC": if input.tf_data_format == "NHWC":
if len(input.out_shapes[0]) == 4: if len(input.out_shapes[0]) == 4:
expand_times = [expand_times[i] for i in [0, 3, 1, 2]] expand_times = [expand_times[i] for i in [0, 3, 1, 2]]
elif len(input.out_shape[0]) == 3: elif len(input.out_shapes[0]) == 3:
expand_times = [expand_times[i] for i in [2, 0, 1]] expand_times = [expand_times[i] for i in [2, 0, 1]]
for i in range(len(expand_times)): for i in range(len(expand_times)):
if expand_times[i] < 0: if expand_times[i] < 0:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册