reduce size of max_input_shape so that the ut can pass on win6 (#40088)

831b69d9 · Sing_chan · GitHub · cac00e0b · 831b69d9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_gather.py ...d/tests/unittests/ir/inference/test_trt_convert_gather.py +1 -1

未找到文件。
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_gather.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_gather.py
@@ -138,7 +138,7 @@ class TrtConvertGatherTest(TrtLayerAutoScanTest):
                    "index_data": [1]
                }
                self.dynamic_shape.max_input_shape = {
-                    "input_data": [128, 256, 128, 256],
+                    "input_data": [128, 256, 64, 128],
                    "index_data": [4]
                }
                self.dynamic_shape.opt_input_shape = {