Searched refs:TRTBaseAllocator (Results 1 – 7 of 7) sorted by relevance
47 class TRTBaseAllocator; variable71 ExecutionContext(TRTBaseAllocator* allocator, void* device_memory, in ExecutionContext()100 TRTBaseAllocator* allocator);105 TRTBaseAllocator* memory_allocator_;120 nvinfer1::ICudaEngine* cuda_engine, TRTBaseAllocator* allocator);
39 class TRTBaseAllocator : public nvinfer1::IGpuAllocator {43 virtual ~TRTBaseAllocator() = default;46 class TRTDeviceAllocator : public TRTBaseAllocator {
163 TRTBaseAllocator* memory_allocator);
216 std::unique_ptr<TRTBaseAllocator> allocator_;
50 nvinfer1::ICudaEngine* cuda_engine, TRTBaseAllocator* allocator) { in Create()
235 TRTBaseAllocator* memory_allocator) { in CreateExecutionContexts()
437 std::unique_ptr<TRTBaseAllocator> trt_allocator; in CreateTRTNode()