Home
last modified time | relevance | path

Searched refs:MS_CTX_ENABLE_MEM_SCHEDULER (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/core/utils/
Dms_context.cc84 set_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER, false); in MsContext()
Dms_context.h75 MS_CTX_ENABLE_MEM_SCHEDULER, enumerator
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc456 auto enable_mem_scheduler = ms_context->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in LoadInputData()
700 auto enable_mem_scheduler = ms_context->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in BuildGraphImpl()
738 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in CompileChildGraph()
2014 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in UpdateOutputTensors()
Dsession_basic.cc1705 auto enable_mem_scheduler = ms_context->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in CreateOutputTensors()
1716 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in UpdateOutputTensors()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc101 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in AssignMemory()
1465 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in LaunchKernelMod()
1536 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in UseMemSchedulerIfNeeded()
/third_party/mindspore/mindspore/ccsrc/vm/
Dtransform.cc594 context_ptr->set_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER, true); in CreateBackend()