Searched refs:ReinitializeDevice (Results 1 – 2 of 2) sorted by relevance
160 void ReinitializeDevice(OpKernelContext* context, PerOpGpuDevice* device,
1031 void BaseGPUDevice::ReinitializeDevice(OpKernelContext* context, in ReinitializeDevice() function in tensorflow::BaseGPUDevice1059 ReinitializeDevice(context, device, stream_id, allocator); in ReinitializeGpuDevice()1061 ReinitializeDevice(context, device, 0, allocator); in ReinitializeGpuDevice()