diff --git a/tests/function_libs/torch_lib/extra_opinfo.py b/tests/function_libs/torch_lib/extra_opinfo.py index b6ccf89b1..1053212ee 100644 --- a/tests/function_libs/torch_lib/extra_opinfo.py +++ b/tests/function_libs/torch_lib/extra_opinfo.py @@ -1767,12 +1767,12 @@ def shape(size, rank, with_batch_channel=True): ) yield opinfo_core.SampleInput( make_arg(shape(D, rank)), - shape(L, rank, False), + None, [1.7, 2.0], ) yield opinfo_core.SampleInput( make_arg(shape(D, rank)), - shape(L, rank, False), + None, [0.6, 0.4], )