Searched defs:AssignMemory (Results 1 – 5 of 5) sorted by relevance
29 int MemoryManager::AssignMemory(const std::vector<std::unique_ptr<OperatorCoder>> &nodes) { in AssignMemory() function in mindspore::lite::micro::MemoryManager
88 void CPUMemoryManager::AssignMemory(const session::KernelGraph *graph) { in AssignMemory() function in mindspore::device::cpu::CPUMemoryManager
256 bool ParameterAggregator::AssignMemory(const K server_kernel, const CNodePtr &cnode, in AssignMemory() function in mindspore::fl::server::ParameterAggregator
436 void GPUKernelRuntime::AssignMemory(const session::KernelGraph &graph) { in AssignMemory() function in mindspore::device::gpu::GPUKernelRuntime
98 void KernelRuntime::AssignMemory(const session::KernelGraph &graph) { in AssignMemory() function in mindspore::device::KernelRuntime