Searched refs:FindMemBlock (Results 1 – 2 of 2) sorted by relevance
50 const auto &mem_block = FindMemBlock(device_addr); in AllocContinuousTensorMem()170 const auto &mem_block = FindMemBlock(mem_buf->device_addr_); in DivideMemBuf()192 DynamicMemBlockPtr DynamicMemPoolBestFit::FindMemBlock(const DeviceMemPtr &device_addr) { in FindMemBlock() function in mindspore::device::DynamicMemPoolBestFit205 const auto &mem_block = FindMemBlock(device_addr); in FreeTensorMem()
135 DynamicMemBlockPtr FindMemBlock(const DeviceMemPtr &device_addr);