Searched refs:partially_replicated_size (Results 1 – 1 of 1) sorted by relevance
918 int64 partially_replicated_size = 1; in GatherParallelDataOperandSharding() local920 partially_replicated_size *= in GatherParallelDataOperandSharding()925 Product(operand_tile_assignment) * partially_replicated_size; in GatherParallelDataOperandSharding()928 partially_replicated_size *= in GatherParallelDataOperandSharding()934 if (partially_replicated_size > 1) { in GatherParallelDataOperandSharding()935 operand_tile_assignment.push_back(partially_replicated_size); in GatherParallelDataOperandSharding()938 return partially_replicated_size > 1 in GatherParallelDataOperandSharding()