Searched refs:operand_tile_assignment (Results 1 – 1 of 1) sorted by relevance
906 std::vector<int64> operand_tile_assignment(gather.operand(0)->shape().rank(), in GatherParallelDataOperandSharding() local915 operand_tile_assignment[operand_dim] = in GatherParallelDataOperandSharding()925 Product(operand_tile_assignment) * partially_replicated_size; in GatherParallelDataOperandSharding()935 operand_tile_assignment.push_back(partially_replicated_size); in GatherParallelDataOperandSharding()937 tile_assignment.Reshape(operand_tile_assignment); in GatherParallelDataOperandSharding()