Home
last modified time | relevance | path

Searched refs:gather_nd (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dgather_nd_op_test.py42 gather_nd_t = array_ops.gather_nd(params, indices)
64 gather_nd_ok_t = array_ops.gather_nd(params, indices_empty)
70 gather_nd_ok_t = array_ops.gather_nd(params, indices_empty)
77 gather_nd_ok_t = array_ops.gather_nd(params_empty, indices_empty)
84 gather_nd_break_t = array_ops.gather_nd(params_empty, indices_nonempty)
95 gather_nd_t = array_ops.gather_nd(params, indices)
105 gather_nd_t = array_ops.gather_nd(params, indices)
115 gather_nd_t = array_ops.gather_nd(params, indices)
129 gather_nd_t = array_ops.gather_nd(params_t, indices)
144 gather_nd_t = array_ops.gather_nd(params_t, indices)
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_gather_nd_op_test.py199 result = ragged_gather_ops.gather_nd(params, indices)
211 ragged_gather_ops.gather_nd(params, indices1)
214 ragged_gather_ops.gather_nd(params, indices2)
236 ragged_gather_ops.gather_nd(params, indices)
Dragged_gather_ops.py339 def gather_nd(params: ragged_tensor.RaggedOrDense, function
387 return array_ops.gather_nd(params, indices, name)
421 result = indices.with_flat_values(gather_nd(params, indices.flat_values))
470 return array_ops.gather_nd(flattened_params, flattened_index_tuples)
481 @dispatch.dispatch_for_api(array_ops.gather_nd)
486 return gather_nd(params, indices, batch_dims, name)
/external/tensorflow/tensorflow/python/ops/
Dsparse_grad.py96 return (None, array_ops.gather_nd(out_grad, sp_indices), None, out_grad)
108 return (None, array_ops.gather_nd(out_grad_reshaped,
254 dense_vals = array_ops.gather_nd(y, scaled_indices)
347 sparse_values_grad = array_ops.gather_nd(grad, sparse_indices)
Darray_grad.py1141 updates_grad = array_ops.gather_nd(grad, indices)
1148 updates_grad = array_ops.gather_nd(grad, indices)
1158 updates_grad = array_ops.gather_nd(grad, indices)
1170 y_output = array_ops.gather_nd(output, indices)
1178 y_grad = array_ops.gather_nd(grad / indicators, indices) * y_indicators
1197 updates_grad = array_ops.gather_nd(grad, indices)
1205 updates_grad = array_ops.gather_nd(grad, indices)
/external/tensorflow/tensorflow/lite/kernels/
Dgather_nd.cc28 namespace gather_nd { namespace
202 gather_nd::Prepare, gather_nd::Eval}; in Register_GATHER_ND()
DAndroid.bp102 "gather_nd.cc",
/external/tensorflow/tensorflow/compiler/tests/
Dwhere_op_test.py51 gathered = array_ops.gather_nd(value, true_vals)
66 gathered = array_ops.gather_nd(value, indices)
Dgather_nd_op_test.py33 gather_nd_t = array_ops.gather_nd(paramsp, indicesp)
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_grad.py42 values=array_ops.gather_nd(grad, coo_sparse_tensor.indices),
92 return (array_ops.gather_nd(unpruned.indices, keep_indices),
93 array_ops.gather_nd(unpruned.values,
182 gather_op = array_ops.gather_nd
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.manip.pbtxt8 name: "gather_nd"
Dtensorflow.-variable.pbtxt96 name: "gather_nd"
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dgather_nd.py63 out = tf.gather_nd(params, indices)
/external/tensorflow/tensorflow/python/distribute/
Dvalues_v2.py219 def gather_nd(self, indices, name=None): member in DistributedVariable
221 return super().gather_nd(indices, name)
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2022-087.md7 …w/blob/f463040eb3997e42e60a2ffc6dc72de7ef11dbb4/tensorflow/lite/kernels/gather_nd.cc#L105-L111) fu…
Dtfsa-2022-143.md7 …/1bc98621180a350eb4e8d3318ea8e228c7559b37/tensorflow/lite/micro/kernels/gather_nd.cc#L143-L154) fu…
Dtfsa-2021-161.md8 …tensorflow/blob/149562d49faa709ea80df1d99fc41d005b81082a/tensorflow/lite/kernels/gather_nd.cc#L124)
/external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/
Darray_ops_test.py66 gather_nd_t = array_ops.gather_nd(params, indices, batch_dims=1)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_GatherV2.pbtxt58 See also `tf.batch_gather` and `tf.gather_nd`.
Dapi_def_GatherNd.pbtxt31 dimension of `params`, in `tf.gather_nd`, `indices` defines slices into the
Dapi_def_ScatterNd.pbtxt35 op is the inverse of the `tf.gather_nd` operator which extracts values or slices
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py75 outputs.append(array_ops.gather_nd(x_i, [0], batch_dims=0))
76 outputs.append(array_ops.gather_nd(x_i, [i], batch_dims=0))
77 outputs.append(array_ops.gather_nd(x_i, [[i], [i], [i]], batch_dims=1))
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable.py125 def gather_nd(self, indices, name=None): member in AutoCastVariable
127 val = self._variable.gather_nd(indices, name=name)
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_grad_test.py92 array_ops.gather_nd(mats_val, indices), dtype=dtypes.float32)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-variable.pbtxt95 name: "gather_nd"

123