Searched refs:IdForHost (Results 1 – 3 of 3) sorted by relevance
82 int IdForHost(TpuDimensionsExternal host) const;
122 int TpuTopologyExternal::IdForHost(TpuDimensionsExternal host) const { in IdForHost() function in tensorflow::tpu::TpuTopologyExternal
158 int task_id = topology.IdForHost(core.host_coordinates()); in GetTpuDevices()