Home
last modified time | relevance | path

Searched refs:set_single_sharding (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_metadata.cc55 instruction->set_single_sharding(sharding); in SetSingleSharding()
Dhlo_instruction.h1164 void set_single_sharding(const HloSharding& sharding);
1167 set_single_sharding(HloSharding::AssignDevice(device)); in set_device_sharding()
Dhlo_instruction.cc1214 void HloInstruction::set_single_sharding(const HloSharding& sharding) { in set_single_sharding() function in xla::HloInstruction