Searched defs:batch_indices (Results 1 – 4 of 4) 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
115 const int64 batch_indices = g.group()[0]; in Compute() local
909 auto batch_indices = in Compute() local
127 auto batch_indices = in ConcatenateIota() local