update reduce ut (#49689)

cfe55f31 · Zhang Jun · GitHub · b854f259 · cfe55f31
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py ...d/tests/unittests/ir/inference/test_trt_convert_reduce.py +3 -0

未找到文件。
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py
@@ -85,6 +85,9 @@ class TrtConvertReduceTest(TrtLayerAutoScanTest):
                            ]
                            dics2 = [
                                {
+                                    "keep_dim": keep_dim,
+                                    "dim": dim,
+                                    "reduce_all": reduce_all,
                                    "out_dtype": out_dtype,
                                    "in_dtype": out_dtype,
                                },