Searched refs:copy_hlo (Results 1 – 1 of 1) sorted by relevance
2445 auto copy_hlo = b->AddInstruction(HloInstruction::CreateUnary( in PartitionDot() local2447 copy_hlo->set_sharding(rhs.sharding()); in PartitionDot()2448 rhs = PartitionedHlo(copy_hlo, rhs.base_shape(), rhs.state()); in PartitionDot()