Searched refs:sharding_ptr (Results 1 – 2 of 2) 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 ()()
1143 std::shared_ptr<const HloSharding> sharding_ptr() const { return sharding_; } in sharding_ptr() function