Home
last modified time | relevance | path

Searched defs:AssignMemory (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_memory_manager.cc91 void CPUMemoryManager::AssignMemory(const session::KernelGraph *graph) { in AssignMemory() function in mindspore::device::cpu::CPUMemoryManager
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/allocator/
Dmemory_manager.cc30 int MemoryManager::AssignMemory(const std::vector<std::unique_ptr<OperatorCoder>> &nodes, in AssignMemory() function in mindspore::lite::micro::MemoryManager
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_kernel_runtime.cc455 void GPUKernelRuntime::AssignMemory(const session::KernelGraph &graph) { in AssignMemory() function in mindspore::device::gpu::GPUKernelRuntime
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc171 void KernelRuntime::AssignMemory(const session::KernelGraph &graph) { in AssignMemory() function in mindspore::device::KernelRuntime