Searched refs:TRTCudaAllocator (Results 1 – 3 of 3) sorted by relevance
49 class TRTCudaAllocator : public TRTBaseAllocator {53 TRTCudaAllocator() {} in TRTCudaAllocator() function54 virtual ~TRTCudaAllocator() {} in ~TRTCudaAllocator()
61 void* TRTCudaAllocator::allocate(uint64_t size, uint64_t alignment, in allocate()69 void TRTCudaAllocator::free(void* memory) { cudaFree(memory); } in free()
655 cres->allocator_.reset(new TRTCudaAllocator); in AllocateCalibrationResources()