Searched defs:dev_ptr (Results 1 – 3 of 3) sorted by relevance
82 void GPUBucket::FreeDeviceMem(void *dev_ptr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(dev_… in FreeDeviceMem()
75 void AscendBucket::FreeDeviceMem(void *dev_ptr) { AscendMemoryPool::GetInstance().FreeTensorMem(dev… in FreeDeviceMem()
83 thrust::device_ptr<T> dev_ptr(array); in ThrustFillWith() local