Searched refs:sharding_ptr (Results 1 – 3 of 3) sorted by relevance
385 sharding = instruction->sharding_ptr(); in ExtractOriginalCommonSharding()476 auto instruction_sharding = instruction->sharding_ptr(); in operator ()()477 auto root_sharding = root->sharding_ptr(); in operator ()()
940 new_instruction->set_sharding(old_instruction->sharding_ptr()); in ReplaceInstruction()
1386 std::shared_ptr<const HloSharding> sharding_ptr() const { return sharding_; } in sharding_ptr() function