Searched refs:base_allocator (Results 1 – 1 of 1) sorted by relevance
754 TfGpuId tf_gpu_id, Allocator* base_allocator, in Reinitialize() argument756 stream_device_.Reinitialize(context, gpu_stream, tf_gpu_id, base_allocator, in Reinitialize()