Home
last modified time | relevance | path

Searched refs:flat_indices (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dscatter_nd_fuzz.cc93 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/compiler/tests/
Dscatter_nd_op_test.py56 flat_indices = _FlatInnerDims(indices)
59 for ix_updates, ix_output in enumerate(flat_indices):
/external/tensorflow/tensorflow/python/ops/
Dctc_ops.py862 flat_indices = math_ops.range(
867 array_ops.boolean_mask(flat_indices, flat_mask), 1)
Darray_ops.py3510 flat_indices = reshape(batch_indices, [-1])
3517 flat_result = gather(flat_params, flat_indices)
/external/tensorflow/tensorflow/python/kernel_tests/
Dscatter_nd_ops_test.py69 flat_indices = _FlatInnerDims(indices)
72 for ix_updates, ix_output in enumerate(flat_indices):