Searched refs:enable_mem_scheduler (Results 1 – 5 of 5) sorted by relevance
456 auto enable_mem_scheduler = ms_context->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in LoadInputData() local457 if (enable_mem_scheduler) { in LoadInputData()700 auto enable_mem_scheduler = ms_context->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in BuildGraphImpl() local701 if (!enable_mem_scheduler) { in BuildGraphImpl()738 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in CompileChildGraph() local739 if (!enable_mem_scheduler) { in CompileChildGraph()2014 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in UpdateOutputTensors() local2015 if (enable_mem_scheduler) { in UpdateOutputTensors()
1705 auto enable_mem_scheduler = ms_context->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in CreateOutputTensors() local1706 if (enable_mem_scheduler) { in CreateOutputTensors()1716 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in UpdateOutputTensors() local1717 if (enable_mem_scheduler) { in UpdateOutputTensors()
101 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in AssignMemory() local102 if (enable_mem_scheduler) { in AssignMemory()1465 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in LaunchKernelMod() local1466 if (enable_mem_scheduler) { in LaunchKernelMod()1536 auto enable_mem_scheduler = context_ptr->get_param<bool>(MS_CTX_ENABLE_MEM_SCHEDULER); in UseMemSchedulerIfNeeded() local1537 if (enable_mem_scheduler) { in UseMemSchedulerIfNeeded()
592 auto enable_mem_scheduler = common::GetEnv(kEnableMemScheduler); in CreateBackend() local593 if (enable_mem_scheduler == "1") { in CreateBackend()
1282 auto enable_mem_scheduler = common::GetEnv(kEnableMemScheduler); in InitHccl()1283 if (single_op == "1" || enable_mem_scheduler == "1") { in InitHccl()