Searched refs:device_memory_allocator_ (Results 1 – 1 of 1) sorted by relevance
75 device_memory_allocator_(nullptr), in ContextDeviceMemory()80 device_memory_allocator_->free(device_memory_); in ~ContextDeviceMemory()88 device_memory_allocator_ = device_memory_allocator; in AllocateDeviceMemory()92 device_memory_ = device_memory_allocator_->allocate( in AllocateDeviceMemory()110 TRTBaseAllocator* device_memory_allocator_; member in tensorflow::tensorrt::__anon58f8ceb50111::ContextDeviceMemory