Home
last modified time | relevance | path

Searched refs:sharding_ptr (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_metadata.cc393 sharding = instruction->sharding_ptr(); in ExtractOriginalCommonSharding()
477 auto instruction_sharding = instruction->sharding_ptr(); in operator ()()
478 auto root_sharding = root->sharding_ptr(); in operator ()()
H A Dconditional_code_motion.cc715 conditional->set_sharding(new_root->sharding_ptr()); in MoveInstructionOut()
868 conditional->set_sharding(new_root->sharding_ptr()); in MoveInstructionIn()
H A Dhlo_computation.cc1010 new_instruction->set_sharding(old_instruction->sharding_ptr()); in ReplaceInstructionWithDifferentShape()
H A Dhlo_instruction.h1553 std::shared_ptr<const HloSharding> sharding_ptr() const { return sharding_; } in sharding_ptr() function