Searched refs:device_alloc (Results 1 – 1 of 1) sorted by relevance
909 auto device_alloc = GetDeviceAndAllocator(params, engine); in ConvertAfterShapes() local911 if (device_alloc.first >= 0) { in ConvertAfterShapes()912 cuda_device_id = device_alloc.first; in ConvertAfterShapes()913 alloc.reset(new TRTDeviceAllocator(device_alloc.second)); in ConvertAfterShapes()