Searched refs:ReinitializeDevice (Results 1 – 2 of 2) sorted by relevance
164 void ReinitializeDevice(OpKernelContext* context, PerOpGpuDevice* device,
930 void BaseGPUDevice::ReinitializeDevice(OpKernelContext* context, in ReinitializeDevice() function in tensorflow::BaseGPUDevice957 ReinitializeDevice(context, device, stream_id, allocator); in ReinitializeGpuDevice()959 ReinitializeDevice(context, device, 0, allocator); in ReinitializeGpuDevice()