Home
last modified time | relevance | path

Searched defs:fusion_threshold (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dparallel_context.cc106 void ParallelContext::set_fusion_threshold_mb(int64_t fusion_threshold) { in set_fusion_threshold_mb()
113 void ParallelContext::set_allgather_fusion_threshold_mb(int64_t fusion_threshold) { in set_allgather_fusion_threshold_mb()
118 void ParallelContext::set_reducescatter_fusion_threshold_mb(int64_t fusion_threshold) { in set_reducescatter_fusion_threshold_mb()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_auto_parallel_context.py171 def set_fusion_threshold_mb(self, fusion_threshold=64, comm_type="allreduce"): argument