Searched defs:pipeline_stages (Results 1 – 3 of 3) sorted by relevance
9 pipeline_stages = 4 variable
3132 int64_t pipeline_stages) { in ReorderForPipelineSplit()3198 auto pipeline_stages = ParallelContext::GetInstance()->pipeline_stage_split_num(); in MicroBatchPreProcess() local3218 auto pipeline_stages = ParallelContext::GetInstance()->pipeline_stage_split_num(); in MicroBatchPostProcess() local3243 auto pipeline_stages = ParallelContext::GetInstance()->pipeline_stage_split_num(); in InsertAllReduceForNormValue() local3675 auto pipeline_stages = ParallelContext::GetInstance()->pipeline_stage_split_num(); in ParallelPartProcess() local3739 auto pipeline_stages = ParallelContext::GetInstance()->pipeline_stage_split_num(); in StepParallel() local
1693 def _check_param_for_integrate_save(pipeline_stages, uniform_split): argument