Searched refs:TpuDimensionsExternal (Results 1 – 3 of 3) sorted by relevance
27 struct TpuDimensionsExternal { struct38 TpuDimensionsExternal chip_coordinates() const; argument39 TpuDimensionsExternal host_coordinates() const;82 int IdForHost(TpuDimensionsExternal host) const;
23 TpuDimensionsExternal TpuCoreLocationExternal::chip_coordinates() const { in chip_coordinates()30 TpuDimensionsExternal TpuCoreLocationExternal::host_coordinates() const { in host_coordinates()122 int TpuTopologyExternal::IdForHost(TpuDimensionsExternal host) const { in IdForHost()
159 const tf_tpu::TpuDimensionsExternal coords = core.chip_coordinates(); in GetTpuDevices()