Searched defs:enable_all_reduce_fusion (Results 1 – 3 of 3) sorted by relevance
34 bool enable_all_reduce_fusion = ParallelContext::GetInstance()->enable_all_reduce_fusion(); in StepAllreduceFusion() local
111 void set_enable_all_reduce_fusion(bool enable_all_reduce_fusion) { in set_enable_all_reduce_fusion()114 bool enable_all_reduce_fusion() const { return enable_all_reduce_fusion_; } in enable_all_reduce_fusion() function
444 def set_enable_all_reduce_fusion(self, enable_all_reduce_fusion): argument