Skip to content

Commit 05b7f11

Browse files
committed
cumsum
Signed-off-by: xavier dupré <[email protected]>
1 parent 1599faf commit 05b7f11

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/test_backend.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5462,7 +5462,7 @@ def func1_length(x):
54625462
self._run_test_case(func1_length, [_OUTPUT], {_INPUT: x_val})
54635463

54645464
@check_tf_min_version("1.14")
5465-
@check_opset_min_version(10, "Slice")
5465+
@check_opset_min_version(11, "CumSum")
54665466
def test_rfft2d_ops(self):
54675467

54685468
x_val = make_xval([3, 4]).astype(np.float32)
@@ -5486,7 +5486,7 @@ def func3(x):
54865486
self._run_test_case(func3, [_OUTPUT], {_INPUT: x_val})
54875487

54885488
@check_tf_min_version("1.14")
5489-
@check_opset_min_version(10, "Slice")
5489+
@check_opset_min_version(11, "CumSum")
54905490
def test_rfft2d_ops_fft_length(self):
54915491

54925492
x_val = make_xval([3, 4]).astype(np.float32)
@@ -5499,7 +5499,7 @@ def func1_length(x):
54995499
self._run_test_case(func1_length, [_OUTPUT], {_INPUT: x_val})
55005500

55015501
@check_tf_min_version("1.14")
5502-
@check_opset_min_version(10, "Slice")
5502+
@check_opset_min_version(11, "CumSum")
55035503
def test_rfft2d_ops_fft_length_many(self):
55045504
for i in range(7, 4, -1):
55055505
for j in range(7, 4, -1):
@@ -5514,7 +5514,7 @@ def func1_length(x):
55145514
self._run_test_case(func1_length, [_OUTPUT], {_INPUT: x_val}, optimize=opt)
55155515

55165516
@check_tf_min_version("1.14")
5517-
@check_opset_min_version(10, "Slice")
5517+
@check_opset_min_version(11, "CumSum")
55185518
@unittest.skipIf(True, reason="Not fully implemented for dynamic shape.")
55195519
def test_fft_ops(self):
55205520
x_val = make_xval([3, 4]).astype(np.float32)
@@ -5538,7 +5538,7 @@ def func(x):
55385538
self._run_test_case(func, [_OUTPUT], {_INPUT: x_val})
55395539

55405540
@check_tf_min_version("1.14")
5541-
@check_opset_min_version(10, "Slice")
5541+
@check_opset_min_version(11, "CumSum")
55425542
def test_rfft2d_ops_specific_dimension(self):
55435543

55445544
x_val = make_xval([3, 1, 4]).astype(np.float32)

0 commit comments

Comments
 (0)