Searched refs:to_adjust_dims (Results 1 – 1 of 1) sorted by relevance
1871 auto to_adjust_dims = operand_parallel_dims; in GatherOperandsShardedAcrossParallelDims() local1876 std::swap(to_adjust_dims, target_dims); in GatherOperandsShardedAcrossParallelDims()1883 for (int i = 0; i < to_adjust_dims.size(); ++i) { in GatherOperandsShardedAcrossParallelDims()1885 int64 to_adjust_dim = to_adjust->tile_assignment().dim(to_adjust_dims[i]); in GatherOperandsShardedAcrossParallelDims()1897 new_tile_assignment_dims[to_adjust_dims[i]] *= ratio; in GatherOperandsShardedAcrossParallelDims()1911 to_adjust_dims, *target, target_dims); in GatherOperandsShardedAcrossParallelDims()1914 to_adjust_dims, *target, target_dims); in GatherOperandsShardedAcrossParallelDims()