Searched defs:stage_num (Results 1 – 6 of 6) sorted by relevance
58 static int64_t InferStage(int64_t rank_id, int64_t stage_num, int64_t device_num) { in InferStage()78 auto stage_num = parallel::ParallelContext::GetInstance()->pipeline_stage_split_num(); in PipelineSplit() local
77 int64_t stage_num() const { return stage_num_; } in stage_num() function
106 void ParallelContext::set_pipeline_stage_split_num(const int64_t stage_num) { pipeline_stage_split_… in set_pipeline_stage_split_num()
52 auto stage_num = parallel::ParallelContext::GetInstance()->pipeline_stage_split_num(); in operator() local119 auto stage_num = parallel::ParallelContext::GetInstance()->pipeline_stage_split_num(); in operator() local
80 auto stage_num = g_device_manager->stage_num(); in IsLastStage() local293 auto stage_num = g_device_manager->stage_num(); in ReorderForForward() local309 auto stage_num = g_device_manager->stage_num(); in ReorderForBackward() local
182 auto stage_num = g_device_manager->stage_num(); in CreateForwardGroup() local228 auto stage_num = g_device_manager->stage_num(); in Coloring() local821 auto stage_num = g_device_manager->stage_num(); in CutBorder() local