Skip to content

Commit f8c254c

Browse files
committed
correct ut
1 parent 71e141f commit f8c254c

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

python/paddle/fluid/tests/unittests/ir/inference/test_trt_reduce_mean_op.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,10 @@ def setUp(self):
6666
self.trt_parameters = TRTReduceMeanAllNoBatchTest.TensorRTParam(
6767
1 << 30, 32, 1, AnalysisConfig.Precision.Float32, False, False)
6868
self.fetch_list = [out]
69-
self.dynamic_shape_params = TRTReduceMeanTest.DynamicShapeParam({
70-
'data': [1, 3, 64, 64]
71-
}, {'data': [3, 3, 224, 224]}, {'data': [3, 3, 224, 224]}, False)
69+
self.dynamic_shape_params = TRTReduceMeanAllNoBatchTest.DynamicShapeParam(
70+
{
71+
'data': [1, 3, 64, 64]
72+
}, {'data': [3, 3, 224, 224]}, {'data': [3, 3, 224, 224]}, False)
7273

7374
def test_check_output(self):
7475
if core.is_compiled_with_cuda():

0 commit comments

Comments
 (0)