未验证 提交 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(
"input_data2": [1]
}
self.dynamic_shape.max_input_shape = {
"input_data1": [256],
"input_data1": [128],
"input_data2": [128]
}
self.dynamic_shape.opt_input_shape = {
"input_data1": [16],
"input_data1": [32],
"input_data2": [32]
}
elif self.dims == 2:
......@@ -232,7 +232,7 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [128, 256]
}
self.dynamic_shape.opt_input_shape = {
"input_data1": [2, 16],
"input_data1": [32, 64],
"input_data2": [32, 64]
}
elif self.dims == 3:
......@@ -241,11 +241,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [1, 4, 4]
}
self.dynamic_shape.max_input_shape = {
"input_data1": [128, 256, 128],
"input_data1": [128, 128, 256],
"input_data2": [128, 128, 256]
}
self.dynamic_shape.opt_input_shape = {
"input_data1": [2, 32, 16],
"input_data1": [2, 64, 64],
"input_data2": [2, 64, 64]
}
elif self.dims == 4:
......@@ -254,11 +254,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
"input_data2": [1, 4, 4, 4]
}
self.dynamic_shape.max_input_shape = {
"input_data1": [8, 32, 64, 64],
"input_data1": [8, 128, 64, 128],
"input_data2": [8, 128, 64, 128]
}
self.dynamic_shape.opt_input_shape = {
"input_data1": [2, 32, 32, 16],
"input_data1": [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.
先完成此消息的编辑!
想要评论请 注册