Searched refs:GetTpuHostLocation (Results 1 – 4 of 4) sorted by relevance
58 virtual const TpuHostLocationExternal GetTpuHostLocation() const = 0;
66 const tensorflow::tpu::TpuHostLocationExternal GetTpuHostLocation()
126 const tensorflow::tpu::TpuHostLocationExternal TpuPlatform::GetTpuHostLocation() in GetTpuHostLocation() function in tensorflow::tpu::TpuPlatform
104 tf_tpu::TpuHostLocationExternal host = platform->GetTpuHostLocation(); in GetDefaultDeviceAssignment()217 int task_id = platform->GetTpuHostLocation().Id(); in GetTpuClient()