Searched defs:root_sharding (Results 1 – 2 of 2) sorted by relevance
477 auto root_sharding = root->sharding_ptr(); in operator ()() local
3403 HloComputation* computation, const HloSharding& root_sharding, in DoPartition()3673 HloComputation* computation, const HloSharding& root_sharding, in PartitionComputation()3718 HloSharding root_sharding = entry_root->sharding(); in Run() local3824 const HloSharding& root_sharding = entry->root_instruction()->sharding(); in PreprocessSharding() local