Searched defs:FreeDeviceMem (Results 1 – 12 of 12) sorted by relevance
87 bool CPUMemoryPool::FreeDeviceMem(const DeviceMemPtr &addr) { in FreeDeviceMem() function in mindspore::device::cpu::CPUMemoryPool
25 void GPULaunchMul::FreeDeviceMem(void *addr) { GPULaunchkernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::gpu::GPULaunchMul
38 void GPULaunchkernel::FreeDeviceMem(void *addr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(a… in FreeDeviceMem() function in mindspore::device::gpu::GPULaunchkernel
98 bool GPUMemoryAllocator::FreeDeviceMem(const DeviceMemPtr &addr) { return CudaDriver::FreeDeviceMem… in FreeDeviceMem() function in mindspore::device::gpu::GPUMemoryAllocator
82 void GPUBucket::FreeDeviceMem(void *dev_ptr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(dev_… in FreeDeviceMem() function in mindspore::device::gpu::GPUBucket
46 bool CudaDriver::FreeDeviceMem(const DeviceMemPtr &addr) { in FreeDeviceMem() function in mindspore::device::gpu::CudaDriver
24 void AscendLaunchKernel::FreeDeviceMem(void *addr) { AscendMemoryPool::GetInstance().FreeTensorMem(… in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchKernel
24 void AscendLaunchMul::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchMul
25 void AscendLaunchTransData::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchTransData
24 void AscendLaunchAtomicClean::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem() function in mindspore::device::ascend::AscendLaunchAtomicClean
75 void AscendBucket::FreeDeviceMem(void *dev_ptr) { AscendMemoryPool::GetInstance().FreeTensorMem(dev… in FreeDeviceMem() function in mindspore::device::ascend::AscendBucket
101 bool AscendMemoryPool::FreeDeviceMem(const DeviceMemPtr &addr) { in FreeDeviceMem() function in mindspore::device::ascend::AscendMemoryPool