Searched refs:SetHandleDeviceErrorCallback (Results 1 – 3 of 3) sorted by relevance
187 void SetHandleDeviceErrorCallback(std::function<Status()> callback);
389 device->SetHandleDeviceErrorCallback(&tpu::TpuNodeContext::CloseTpuHost); in CreateDevices()
535 void XlaDevice::SetHandleDeviceErrorCallback(std::function<Status()> callback) { in SetHandleDeviceErrorCallback() function in tensorflow::XlaDevice