未验证 提交 113539eb 编写于 作者: W Wangzheee 提交者: GitHub

tem_fix_reshape_unitest (#34069)

上级 e2e1c57b
...@@ -99,9 +99,11 @@ class TRTReshapeTest2(TRTReshapeTest): ...@@ -99,9 +99,11 @@ class TRTReshapeTest2(TRTReshapeTest):
self.enable_trt = True self.enable_trt = True
self.trt_parameters = TRTReshapeTest.TensorRTParam( self.trt_parameters = TRTReshapeTest.TensorRTParam(
1 << 30, self.bs, 1, AnalysisConfig.Precision.Float32, False, False) 1 << 30, self.bs, 1, AnalysisConfig.Precision.Float32, False, False)
'''
self.dynamic_shape_params = TRTReshapeTest.DynamicShapeParam({ self.dynamic_shape_params = TRTReshapeTest.DynamicShapeParam({
'data': [1, 3, 8, 8] 'data': [1, 3, 8, 8]
}, {'data': [5, 100, 100, 100]}, {'data': [1, 3, 16, 16]}, False) }, {'data': [5, 100, 100, 100]}, {'data': [1, 3, 16, 16]}, False)
'''
self.fetch_list = [out] self.fetch_list = [out]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册