| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_nd_op_test.py | 45 gather_nd_t = array_ops.gather_nd(params, indices)
67 gather_nd_ok_t = array_ops.gather_nd(params, indices_empty)
73 gather_nd_ok_t = array_ops.gather_nd(params, indices_empty)
80 gather_nd_ok_t = array_ops.gather_nd(params_empty, indices_empty)
87 gather_nd_break_t = array_ops.gather_nd(params_empty, indices_nonempty)
98 gather_nd_t = array_ops.gather_nd(params, indices)
108 gather_nd_t = array_ops.gather_nd(params, indices)
118 gather_nd_t = array_ops.gather_nd(params, indices)
132 gather_nd_t = array_ops.gather_nd(params_t, indices)
147 gather_nd_t = array_ops.gather_nd(params_t, indices)
[all …]
|
| D | lu_op_test.py | 115 permuted_verification_reshaped = array_ops.gather_nd(
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_nd_op_test.py | 203 result = ragged_gather_ops.gather_nd(params, indices)
215 ragged_gather_ops.gather_nd(params, indices1)
218 ragged_gather_ops.gather_nd(params, indices2)
240 ragged_gather_ops.gather_nd(params, indices)
|
| D | ragged_gather_ops.py | 330 def gather_nd(params, indices, batch_dims=0, name=None): function
375 return array_ops.gather_nd(params, indices, name)
409 result = indices.with_flat_values(gather_nd(params, indices.flat_values))
458 return array_ops.gather_nd(flattened_params, flattened_index_tuples)
|
| D | ragged_dispatch.py | 429 return ragged_gather_ops.gather_nd(
489 (array_ops.gather_nd, _ragged_gather_nd_v1, ['params', 'indices']),
490 (array_ops.gather_nd_v2, ragged_gather_ops.gather_nd, ['params',
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 103 return (None, array_ops.gather_nd(out_grad, sp_indices), None, out_grad)
115 return (None, array_ops.gather_nd(out_grad_reshaped, sp_indices // scale),
227 dense_vals = array_ops.gather_nd(y, scaled_indices)
319 sparse_values_grad = array_ops.gather_nd(grad, sparse_indices)
|
| D | array_grad.py | 1115 updates_grad = array_ops.gather_nd(grad, indices)
1122 updates_grad = array_ops.gather_nd(grad, indices)
1132 updates_grad = array_ops.gather_nd(grad, indices)
1144 y_output = array_ops.gather_nd(output, indices)
1152 y_grad = array_ops.gather_nd(grad / indicators, indices) * y_indicators
1171 updates_grad = array_ops.gather_nd(grad, indices)
1179 updates_grad = array_ops.gather_nd(grad, indices)
|
| D | array_ops.py | 5196 def gather_nd(params, indices, name=None, batch_dims=0): function
5350 return params.gather_nd(indices, name=name)
5352 return gen_array_ops.gather_nd(params, indices, name=name)
5360 return gather_nd(params, indices, name=name, batch_dims=batch_dims)
5363 gather_nd_v2.__doc__ = gather_nd.__doc__
5435 out = gen_array_ops.gather_nd(params, indices)
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 27 namespace gather_nd { namespace
174 gather_nd::Prepare, gather_nd::Eval}; in Register_GATHER_ND()
|
| D | Android.bp | 98 "gather_nd.cc",
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | where_op_test.py | 55 gathered = array_ops.gather_nd(value, true_vals)
70 gathered = array_ops.gather_nd(value, indices)
|
| D | gather_nd_op_test.py | 37 gather_nd_t = array_ops.gather_nd(paramsp, indicesp)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.manip.pbtxt | 8 name: "gather_nd"
|
| D | tensorflow.-variable.pbtxt | 96 name: "gather_nd"
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | gather_nd.py | 61 out = tf.gather_nd(params, indices)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 78 outputs.append(array_ops.gather_nd(x_i, [0], batch_dims=0))
79 outputs.append(array_ops.gather_nd(x_i, [i], batch_dims=0))
80 outputs.append(array_ops.gather_nd(x_i, [[i], [i], [i]], batch_dims=1))
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GatherV2.pbtxt | 58 See also `tf.batch_gather` and `tf.gather_nd`.
|
| D | api_def_GatherNd.pbtxt | 31 dimension of `params`, in `tf.gather_nd`, `indices` defines slices into the
|
| D | api_def_ScatterNd.pbtxt | 33 `tf.gather_nd` operator which extracts values or slices from a given tensor.
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 130 def gather_nd(self, indices, name=None): member in AutoCastVariable
132 val = self._variable.gather_nd(indices, name=name)
|
| D | autocast_variable_test.py | 107 self.assertEqual(x.gather_nd([0]).dtype, dtypes.float32)
111 self.assertEqual(x.gather_nd([0]).dtype, dtypes.float16)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 95 name: "gather_nd"
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 80 "kernels/gather_nd.cc",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 81 from tensorflow.lite.testing.op_tests.gather_nd import make_gather_nd_tests
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 1000 permuted_rhs = array_ops.gather_nd(broadcast_rhs, broadcast_perm)
1142 x = array_ops.gather_nd(x, array_ops.stack([batch_indices, perm],
|