Searched refs:input_indices_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 587 auto input_indices_t = input_tensor.indices().matrix<int64>(); in Split() local
627 const int dim = input_indices_t(i, split_dim); in Split()
632 const int64 original_dim = input_indices_t(i, j); in Split()
668 auto input_indices_t = input_tensor.indices().matrix<int64>(); in Slice() local
683 if (!(start[dim] <= input_indices_t(i, dim) && in Slice()
684 input_indices_t(i, dim) < start[dim] + size[dim])) { in Slice()
710 if (!(start[dim] <= input_indices_t(i, dim) && in Slice()
711 input_indices_t(i, dim) < start[dim] + size[dim])) { in Slice()
721 output_indices_t(index, dim) = input_indices_t(i, dim) - start[dim]; in Slice()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op_gpu.h | 446 auto input_indices_t = To32Bit(input_indices.flat<int32>());
447 input_indices_t.device(d) =
448 input_indices_t.generate(ColumnIndexCreator(num_cols));
479 /* d_values_in */ input_indices_t.data(),
504 /* d_values_in */ input_indices_t.data(),
|