Home
last modified time | relevance | path

Searched refs:TRTCudaAllocator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_allocator.h49 class TRTCudaAllocator : public TRTBaseAllocator {
53 TRTCudaAllocator() {} in TRTCudaAllocator() function
54 virtual ~TRTCudaAllocator() {} in ~TRTCudaAllocator()
Dtrt_allocator.cc61 void* TRTCudaAllocator::allocate(uint64_t size, uint64_t alignment, in allocate()
69 void TRTCudaAllocator::free(void* memory) { cudaFree(memory); } in free()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc655 cres->allocator_.reset(new TRTCudaAllocator); in AllocateCalibrationResources()