Searched refs:flat_group_index (Results 1 – 1 of 1) sorted by relevance
403 void PopulateGroupIndices(const int64 flat_group_index, VarDimArray group_shape, in PopulateGroupIndices() argument406 int64 running_flat_group_index = flat_group_index; in PopulateGroupIndices()451 for (int64 flat_group_index = 0; flat_group_index < num_elements; in ComputeDenseToDense() local452 ++flat_group_index) { in ComputeDenseToDense()453 PopulateGroupIndices(flat_group_index, group_shape, &group_indices); in ComputeDenseToDense()510 for (int64 flat_group_index = 0; flat_group_index < num_elements; in ComputeDenseToSparse() local511 ++flat_group_index) { in ComputeDenseToSparse()512 PopulateGroupIndices(flat_group_index, group_shape, &group_indices); in ComputeDenseToSparse()