Searched refs:TRTDeviceAllocator (Results 1 – 4 of 4) sorted by relevance
58 void* TRTDeviceAllocator::allocate(uint64_t size, uint64_t alignment, in allocate()92 TRTDeviceAllocator::TRTDeviceAllocator(Allocator* allocator) in TRTDeviceAllocator() function in tensorflow::tensorrt::TRTDeviceAllocator97 void TRTDeviceAllocator::free(void* memory) { in free()
46 class TRTDeviceAllocator : public TRTBaseAllocator {49 TRTDeviceAllocator(Allocator* allocator);53 virtual ~TRTDeviceAllocator() { in ~TRTDeviceAllocator()
69 allocator_.reset(new TRTDeviceAllocator(alloc)); in TRTEngineCacheResource()
440 trt_allocator.reset(new TRTDeviceAllocator(device_allocator.second)); in CreateTRTNode()