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