Searched refs:parallel_context (Results 1 – 6 of 6) sorted by relevance
226 auto parallel_context = parallel::ParallelContext::GetInstance(); in GetRank() local227 MS_EXCEPTION_IF_NULL(parallel_context); in GetRank()228 if (parallel_context->parallel_mode() != parallel::STAND_ALONE) { in GetRank()250 auto parallel_context = parallel::ParallelContext::GetInstance(); in IsStandAlone() local251 MS_EXCEPTION_IF_NULL(parallel_context); in IsStandAlone()252 return parallel_context->parallel_mode() == parallel::STAND_ALONE; in IsStandAlone()
61 auto parallel_context = parallel::ParallelContext::GetInstance(); in ObtainMulInputsAddr() local62 MS_EXCEPTION_IF_NULL(parallel_context); in ObtainMulInputsAddr()63 auto device_num = parallel_context->device_num(); in ObtainMulInputsAddr()
87 auto parallel_context = parallel::ParallelContext::GetInstance(); in CalculateMean() local88 MS_EXCEPTION_IF_NULL(parallel_context); in CalculateMean()89 auto grad_mean = parallel_context->gradients_mean(); in CalculateMean()
155 auto parallel_context = parallel::ParallelContext::GetInstance(); in GetSplitSegments() local156 MS_EXCEPTION_IF_NULL(parallel_context); in GetSplitSegments()158 if (!parallel_context->enable_parallel_optimizer()) { in GetSplitSegments()159 split_indices = parallel_context->GetAllReduceFusionSplitIndices(group); in GetSplitSegments()
2389 auto parallel_context = parallel::ParallelContext::GetInstance(); in GetAllReduceSplitIndex() local2390 MS_EXCEPTION_IF_NULL(parallel_context); in GetAllReduceSplitIndex()2393 return parallel_context->GetAllReduceFusionSplitIndices(group); in GetAllReduceSplitIndex()2470 auto parallel_context = parallel::ParallelContext::GetInstance(); in InitAllBucket() local2471 MS_EXCEPTION_IF_NULL(parallel_context); in InitAllBucket()2472 auto parallel_mode = parallel_context->parallel_mode(); in InitAllBucket()2512 auto parallel_context = parallel::ParallelContext::GetInstance(); in AddGradAddrToBucket() local2513 MS_EXCEPTION_IF_NULL(parallel_context); in AddGradAddrToBucket()2514 auto parallel_mode = parallel_context->parallel_mode(); in AddGradAddrToBucket()
757 const auto ¶llel_context = parallel::ParallelContext::GetInstance(); in CheckPyNativeContext() local758 MS_EXCEPTION_IF_NULL(parallel_context); in CheckPyNativeContext()761 const auto ¶llel_mode = parallel_context->parallel_mode(); in CheckPyNativeContext()