From f6c11a704eeab2fc90edd66797086ca15f88d625 Mon Sep 17 00:00:00 2001 From: Nripesh Niketan <86844847+NripeshN@users.noreply.github.com> Date: Wed, 7 Feb 2024 14:58:09 +0400 Subject: [PATCH] Revert "feat: Add tensorflow GatherNd raw_ops" (#28210) --- .../frontends/tensorflow/raw_ops.py | 5 --- .../test_tensorflow/test_raw_ops.py | 36 ------------------- 2 files changed, 41 deletions(-) diff --git a/ivy/functional/frontends/tensorflow/raw_ops.py b/ivy/functional/frontends/tensorflow/raw_ops.py index 89367fe724f53..1db6e68e803af 100644 --- a/ivy/functional/frontends/tensorflow/raw_ops.py +++ b/ivy/functional/frontends/tensorflow/raw_ops.py @@ -619,11 +619,6 @@ def Gather(*, params, indices, validate_indices=None, name="Gather"): return ivy.gather(params, indices, axis=0, batch_dims=0) -@to_ivy_arrays_and_back -def GatherNd(*, params, indices, validate_indices=None, name="GatherNd"): - return ivy.gather_nd(params, indices, batch_dims=0) - - @to_ivy_arrays_and_back def Greater(*, x, y, name="Greater"): x, y = check_tensorflow_casting(x, y) diff --git a/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_raw_ops.py b/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_raw_ops.py index 62cdde61ce9df..027d7b72c34db 100644 --- a/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_raw_ops.py +++ b/ivy_tests/test_ivy/test_frontends/test_tensorflow/test_raw_ops.py @@ -2132,42 +2132,6 @@ def test_tensorflow_Gather( # NOQA ) -# GatherNd -@handle_frontend_test( - fn_tree="tensorflow.raw_ops.GatherNd", - params_indices_axis_batch_dims=helpers.array_indices_axis( - array_dtypes=helpers.get_dtypes("valid"), - indices_dtypes=["int64"], - min_num_dims=5, - max_num_dims=10, - min_dim_size=1, - max_dim_size=5, - indices_same_dims=False, - ), - test_with_out=st.just(False), -) -def test_tensorflow_GatherNd( - *, - params_indices_axis_batch_dims, - frontend, - test_flags, - fn_tree, - backend_fw, - on_device, -): - input_dtypes, params, indices, axis, batch_dims = params_indices_axis_batch_dims - helpers.test_frontend_function( - input_dtypes=input_dtypes, - backend_to_test=backend_fw, - frontend=frontend, - test_flags=test_flags, - fn_tree=fn_tree, - on_device=on_device, - params=params, - indices=indices, - ) - - # Greater @handle_frontend_test( fn_tree="tensorflow.raw_ops.Greater",