Searched refs:PreprocessSharding (Results 1 – 2 of 2) sorted by relevance
244 Status PreprocessSharding(HloModule* module);
3652 TF_RETURN_IF_ERROR(PreprocessSharding(module)); in Run()3733 Status SpmdPartitioner::PreprocessSharding(HloModule* module) { in PreprocessSharding() function in xla::spmd::SpmdPartitioner