Searched refs:CloseTpuHost (Results 1 – 6 of 6) sorted by relevance
58 static Status CloseTpuHost();
55 Status TpuNodeContext::CloseTpuHost() { in CloseTpuHost() function in tensorflow::tpu::TpuNodeContext
68 Status close_status = TpuNodeContext::CloseTpuHost(); in Compute()
100 TF_CHECK_OK(tpu::TpuNodeContext::CloseTpuHost()); in Cancel()
311 Status status = TpuNodeContext::CloseTpuHost(); in TPUCancelExecution()
389 device->SetHandleDeviceErrorCallback(&tpu::TpuNodeContext::CloseTpuHost); in CreateDevices()