Searched refs:set_single_sharding (Results 1 – 3 of 3) sorted by relevance
55 instruction->set_single_sharding(sharding); in SetSingleSharding()
1407 void set_single_sharding(const HloSharding& sharding);1410 set_single_sharding(HloSharding::AssignDevice(device)); in set_device_sharding()
1498 void HloInstruction::set_single_sharding(const HloSharding& sharding) { in set_single_sharding() function in xla::HloInstruction