From 0c37fdb5530755827bd4e4da17c594ddf776d179 Mon Sep 17 00:00:00 2001 From: Jason Date: Fri, 17 Apr 2020 14:43:40 +0800 Subject: [PATCH] Update tf_op_mapper.py --- x2paddle/op_mapper/tf_op_mapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/tf_op_mapper.py b/x2paddle/op_mapper/tf_op_mapper.py index 45fd2e8..4d22d02 100644 --- a/x2paddle/op_mapper/tf_op_mapper.py +++ b/x2paddle/op_mapper/tf_op_mapper.py @@ -718,7 +718,7 @@ class TFOpMapper(OpMapper): if input.tf_data_format == "NHWC": if len(input.out_shapes[0]) == 4: 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]] for i in range(len(expand_times)): if expand_times[i] < 0: -- GitLab