Searched defs:comm_type (Results 1 – 1 of 1) sorted by relevance
171 def set_fusion_threshold_mb(self, fusion_threshold=64, comm_type="allreduce"): argument1105 def _set_allgather_comm_fusion(self, comm_fusion, comm_type="allgather"): argument