Searched refs:AssignKernelAddress (Results 1 – 5 of 5) sorted by relevance
40 void AssignKernelAddress(session::KernelGraph *kernel_graph);
62 void CPUKernelRuntime::AssignKernelAddress(session::KernelGraph *kernel_graph) { in AssignKernelAddress() function in mindspore::device::cpu::CPUKernelRuntime
148 runtime_.AssignKernelAddress(graph.get()); in CompileGraphImpl()295 runtime_.AssignKernelAddress(kernel_graph.get()); in RunOpImpl()
149 …void AssignKernelAddress(const std::shared_ptr<MemScheduler> &mem_scheduler, const AnfNodePtr &ker…
1298 void KernelRuntime::AssignKernelAddress(const std::shared_ptr<MemScheduler> &mem_scheduler, const A… in AssignKernelAddress() function in mindspore::device::KernelRuntime1435 AssignKernelAddress(mem_scheduler, kernel, &kernel_inputs, &kernel_workspaces, &kernel_outputs); in LaunchKernel()