Searched refs:batch_indices (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op_gpu.cu.cc | 33 const Eigen::array<int64, IXDIM> batch_indices, const int64 indices_size, in GatherSliceOpKernel() argument
45 out_of_bounds |= !FastBoundsCheck(index_j, batch_indices[j]); in GatherSliceOpKernel()
78 Eigen::array<int64, IXDIM> batch_indices; in operator ()() local
81 batch_indices[size_t(IXDIM - 1)] = Tparams.dimension(IXDIM - 1); in operator ()()
84 batch_indices[i - 1] = Tparams.dimension(i - 1); in operator ()()
92 Tout.data(), batch_strides, batch_indices, in operator ()()
|
| D | ctc_loss_op.cc | 115 const int64 batch_indices = g.group()[0]; in Compute() local
116 OP_REQUIRES(ctx, FastBoundsCheck(batch_indices, batch_size), in Compute()
119 " but saw: ", batch_indices)); in Compute()
122 std::vector<int>* b_values = &labels_t[batch_indices]; in Compute()
|
| D | topk_op_gpu.h | 360 auto batch_indices = indices + offset;
367 batch_indices);
|
| D | batch_kernels.cc | 909 auto batch_indices = in Compute() local
912 sizes.push_back(batch_indices(i, 2) - batch_indices(i, 1)); in Compute()
913 batch_keys.push_back(batch_indices(i, 0)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | data_feeder.py | 534 batch_indices = self.indices[self.offset:end]
538 dict([(self._input_placeholder[k].name, extract(v, batch_indices))
541 extract(self._x, batch_indices)
564 assign_label(v, shape, dtype, n_classes, batch_indices)
571 assign_label(self._y, shape, dtype, n_classes, batch_indices)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lu_op_test.py | 111 batch_indices = math_ops.cast(
117 array_ops.stack([batch_indices, inv_perm_reshaped], axis=-1))
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 127 auto batch_indices = in ConcatenateIota() local
131 return xla::ConcatInDim(b, {batch_indices, indices}, dimensions.size() - 1); in ConcatenateIota()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3494 batch_indices = indices
3508 batch_indices += reshape(dim_indices, dim_shape)
3510 flat_indices = reshape(batch_indices, [-1])
|