Searched refs:TRTDeviceAllocator (Results 1 – 5 of 5) sorted by relevance
71 void* TRTDeviceAllocator::allocate(uint64_t size, uint64_t alignment, in allocate()97 TRTDeviceAllocator::TRTDeviceAllocator(Allocator* allocator) in TRTDeviceAllocator() function in tensorflow::tensorrt::TRTDeviceAllocator102 void TRTDeviceAllocator::free(void* memory) { in free()
59 class TRTDeviceAllocator : public TRTBaseAllocator {62 TRTDeviceAllocator(Allocator* allocator);66 virtual ~TRTDeviceAllocator() { in ~TRTDeviceAllocator()
153 allocator_.reset(new TRTDeviceAllocator(alloc)); in TRTEngineCacheResource()
657 cres->allocator_.reset(new TRTDeviceAllocator(alloc)); in AllocateCalibrationResources()
913 alloc.reset(new TRTDeviceAllocator(device_alloc.second)); in ConvertAfterShapes()