Searched defs:target_dim (Results 1 – 2 of 2) sorted by relevance
1758 int64_t target_dim = target->tile_assignment().dim(target_dims[i]); in GatherOperandsShardedAcrossParallelDims() local
1496 int64_t target_dim = source_target_dims[0].second; in ReshardWithAllToAll() local