Searched defs:device_addr (Results 1 – 4 of 4) sorted by relevance
33 DeviceMemPtr device_addr = FindIdleMemBuf(align_size); in AllocTensorMem() local44 auto device_addr = AllocTensorMem(total_size); in AllocContinuousTensorMem() local116 DeviceMemPtr device_addr = nullptr; in AddMemBlockAndMemBuf() local186 bool DynamicMemPoolBestFit::CmpMemBlock(const DeviceMemPtr &device_addr, const DynamicMemBlockPtr &… in CmpMemBlock()192 DynamicMemBlockPtr DynamicMemPoolBestFit::FindMemBlock(const DeviceMemPtr &device_addr) { in FindMemBlock()202 void DynamicMemPoolBestFit::FreeTensorMem(const DeviceMemPtr &device_addr) { in FreeTensorMem()214 …mPoolBestFit::CombineMemBuf(const DynamicMemBlockPtr &mem_block, const DeviceMemPtr &device_addr) { in CombineMemBuf()266 void DynamicMemPoolBestFit::EraseIdleMemBuf(size_t size, const DeviceMemPtr &device_addr) { in EraseIdleMemBuf()286 auto &device_addr = (*iter)->device_addr_base_; in ReleaseDeviceRes() local
66 const DeviceMemPtr &device_addr() const { return device_addr_base_; } in device_addr() function
81 auto device_addr = MallocMemFromMemPool(1); in MallocDeviceMemory() local
639 auto device_addr = AnfAlgo::GetOutputAddr(node, index); in CreateSomasParameter() local