| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_nd_op_test.py | 43 gather_nd_t = array_ops.gather_nd(params, indices)
65 gather_nd_ok_t = array_ops.gather_nd(params, indices_empty)
71 gather_nd_ok_t = array_ops.gather_nd(params, indices_empty)
78 gather_nd_ok_t = array_ops.gather_nd(params_empty, indices_empty)
85 gather_nd_break_t = array_ops.gather_nd(params_empty, indices_nonempty)
96 gather_nd_t = array_ops.gather_nd(params, indices)
106 gather_nd_t = array_ops.gather_nd(params, indices)
116 gather_nd_t = array_ops.gather_nd(params, indices)
130 gather_nd_t = array_ops.gather_nd(params_t, indices)
145 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_ops.py | 125 def gather_nd(params, indices, name=None): function
169 return array_ops.gather_nd(params, indices, name)
201 result = indices.with_flat_values(gather_nd(params, indices.flat_values))
250 return array_ops.gather_nd(flattened_params, flattened_index_tuples)
|
| D | ragged_gather_nd_op_test.py | 204 result = ragged_gather_ops.gather_nd(params, indices)
216 ragged_gather_ops.gather_nd(params, indices1)
219 ragged_gather_ops.gather_nd(params, indices2)
241 ragged_gather_ops.gather_nd(params, indices)
|
| D | ragged_dispatch.py | 436 (array_ops.gather_nd, ragged_gather_ops.gather_nd, ['params', 'indices']),
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_GatherNd.pbtxt | 4 name: "gather_nd"
7 name: "manip.gather_nd"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 25 namespace gather_nd { namespace
149 gather_nd::Prepare, gather_nd::Eval}; in Register_GATHER_ND()
|
| D | Android.bp | 69 "gather_nd.cc",
|
| /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 | 865 updates_grad = array_ops.gather_nd(grad, indices)
872 updates_grad = array_ops.gather_nd(grad, indices)
882 updates_grad = array_ops.gather_nd(grad, indices)
890 updates_grad = array_ops.gather_nd(grad, indices)
898 updates_grad = array_ops.gather_nd(grad, indices)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | sampler.py | 358 sample_ids_sampling = array_ops.gather_nd(sample_ids, where_sampling)
359 inputs_not_sampling = array_ops.gather_nd(base_next_inputs,
459 auxiliary_inputs = array_ops.gather_nd(auxiliary_inputs, indices)
471 outputs_sampling = array_ops.gather_nd(outputs, where_sampling)
472 inputs_not_sampling = array_ops.gather_nd(base_next_inputs,
|
| D | helper.py | 398 sample_ids_sampling = array_ops.gather_nd(sample_ids, where_sampling)
399 inputs_not_sampling = array_ops.gather_nd(
518 auxiliary_inputs = array_ops.gather_nd(auxiliary_inputs, indices)
532 outputs_sampling = array_ops.gather_nd(outputs, where_sampling)
533 inputs_not_sampling = array_ops.gather_nd(base_next_inputs,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.manip.pbtxt | 8 name: "gather_nd"
|
| /external/tensorflow/tensorflow/contrib/sparsemax/python/ops/ |
| D | sparsemax.py | 81 tau_sum = array_ops.gather_nd(z_cumsum, indices)
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| D | losses.py | 126 array_ops.gather_nd(params=logits, indices=indices),
|
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| D | sparse_ops.py | 77 values=array_ops.gather_nd(dense_tensor, indices, name="values"),
|
| /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/contrib/crf/python/ops/ |
| D | crf.py | 94 sequence_scores = array_ops.gather_nd(
513 gen_array_ops.gather_nd(inputs, indices), # [B]
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | composite_tensor_support_test.py | 76 values = array_ops.gather_nd(inputs, indices)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | gather_nd_op_test.py | 36 gather_nd_t = array_ops.gather_nd(paramsp, indicesp)
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_optimizer.py | 128 sparse_values = array_ops.gather_nd(dense_tensor, sparse_indices)
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 74 "kernels/gather_nd.cc",
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_colorbot/ |
| D | rnn_colorbot.py | 204 hidden_states = tf.gather_nd(chars, indices)
|