Searched defs:batch_indices (Results 1 – 5 of 5) 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
153 const int64 batch_indices = g.group()[0]; in Compute() local
588 auto batch_indices = in Compute() local
107 const std::vector<int64>& batch_indices(int i) const { in batch_indices() function
127 auto batch_indices = in ConcatenateIota() local