未验证 提交 f50768e8 编写于 作者: Z zlsh80826 提交者: GitHub

Revise input_data1 shape to same as input_data2 shape for non-broadcast cases (#38206)

上级 ced6ab6d
...@@ -215,11 +215,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast( ...@@ -215,11 +215,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [1] "input_data2": [1]
} }
self.dynamic_shape.max_input_shape = { self.dynamic_shape.max_input_shape = {
"input_data1": [256], "input_data1": [128],
"input_data2": [128] "input_data2": [128]
} }
self.dynamic_shape.opt_input_shape = { self.dynamic_shape.opt_input_shape = {
"input_data1": [16], "input_data1": [32],
"input_data2": [32] "input_data2": [32]
} }
elif self.dims == 2: elif self.dims == 2:
...@@ -232,7 +232,7 @@ class TrtConvertElementwiseTest_two_input_without_broadcast( ...@@ -232,7 +232,7 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [128, 256] "input_data2": [128, 256]
} }
self.dynamic_shape.opt_input_shape = { self.dynamic_shape.opt_input_shape = {
"input_data1": [2, 16], "input_data1": [32, 64],
"input_data2": [32, 64] "input_data2": [32, 64]
} }
elif self.dims == 3: elif self.dims == 3:
...@@ -241,11 +241,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast( ...@@ -241,11 +241,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [1, 4, 4] "input_data2": [1, 4, 4]
} }
self.dynamic_shape.max_input_shape = { self.dynamic_shape.max_input_shape = {
"input_data1": [128, 256, 128], "input_data1": [128, 128, 256],
"input_data2": [128, 128, 256] "input_data2": [128, 128, 256]
} }
self.dynamic_shape.opt_input_shape = { self.dynamic_shape.opt_input_shape = {
"input_data1": [2, 32, 16], "input_data1": [2, 64, 64],
"input_data2": [2, 64, 64] "input_data2": [2, 64, 64]
} }
elif self.dims == 4: elif self.dims == 4:
...@@ -254,11 +254,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast( ...@@ -254,11 +254,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [1, 4, 4, 4] "input_data2": [1, 4, 4, 4]
} }
self.dynamic_shape.max_input_shape = { self.dynamic_shape.max_input_shape = {
"input_data1": [8, 32, 64, 64], "input_data1": [8, 128, 64, 128],
"input_data2": [8, 128, 64, 128] "input_data2": [8, 128, 64, 128]
} }
self.dynamic_shape.opt_input_shape = { self.dynamic_shape.opt_input_shape = {
"input_data1": [2, 32, 32, 16], "input_data1": [2, 64, 32, 32],
"input_data2": [2, 64, 32, 32] "input_data2": [2, 64, 32, 32]
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册