Searched defs:batch_indices (Results 1 – 7 of 7) sorted by relevance
33 const Eigen::array<int64, IXDIM> batch_indices, const int64 indices_size, in GatherSliceOpKernel()78 Eigen::array<int64, IXDIM> batch_indices; in operator ()() local
162 const int64_t batch_indices = g.group()[0]; in Compute() local
362 auto batch_indices = indices + offset; in TopKKernel() local
674 auto batch_indices = in Compute() local
107 const std::vector<int64_t>& batch_indices(int i) const { in batch_indices() function
130 auto batch_indices = in ConcatenateIota() local
633 Output batch_indices = indices; in GetBatchIndices() local