Searched refs:flat_indices (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 93 auto flat_indices = indices_tensor.flat<int32>(); in FuzzImpl() local
95 flat_indices(i) = data[data_ix++]; in FuzzImpl()
98 flat_indices(i) = 0; // ensure that indices_tensor has all values in FuzzImpl()
|
| /external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/ |
| D | scatter_nd_ops_test.py | 56 flat_indices = _FlatInnerDims(indices)
59 for ix_updates, ix_output in enumerate(flat_indices):
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 490 flat_indices = array_ops.reshape(indices, [-1])
494 array_ops.gather(combined_splits, flat_indices),
495 array_ops.gather(combined_splits[1:], flat_indices)).values
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 57 flat_indices = _FlatInnerDims(indices)
60 for ix_updates, ix_output in enumerate(flat_indices):
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 1141 flat_indices = math_ops.range(
1146 array_ops.boolean_mask(flat_indices, flat_mask), 1)
|
| D | array_ops.py | 5176 flat_indices = reshape(batch_indices, [-1])
5183 flat_result = gather(flat_params, flat_indices)
5431 flat_indices = reshape(indices, shape=flat_shape)
5433 indices = concat((index_grid, flat_indices), axis=-1)
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 70 flat_indices = _FlatInnerDims(indices)
73 for ix_updates, ix_output in enumerate(flat_indices):
|