Searched refs:to_adjust_dim (Results 1 – 1 of 1) sorted by relevance
1885 int64 to_adjust_dim = to_adjust->tile_assignment().dim(to_adjust_dims[i]); in GatherOperandsShardedAcrossParallelDims() local1886 if (target_dim < to_adjust_dim) { in GatherOperandsShardedAcrossParallelDims()1889 if (target_dim == to_adjust_dim) { in GatherOperandsShardedAcrossParallelDims()1892 int64 ratio = target_dim / to_adjust_dim; in GatherOperandsShardedAcrossParallelDims()1893 if (target_dim % to_adjust_dim != 0 || in GatherOperandsShardedAcrossParallelDims()