Searched defs:root_sharding (Results 1 – 2 of 2) sorted by relevance
478 auto root_sharding = root->sharding_ptr(); in operator ()() local
4191 HloComputation* computation, const HloSharding& root_sharding, in DoPartition()4462 HloComputation* computation, const HloSharding& root_sharding, in PartitionComputation()4512 HloSharding root_sharding = entry_root->sharding(); in Run() local4630 const HloSharding& root_sharding = entry->root_instruction()->sharding(); in PreprocessSharding() local