Searched refs:TpuHostLocationExternal (Results 1 – 8 of 8) sorted by relevance
49 class TpuHostLocationExternal {51 explicit TpuHostLocationExternal(SE_TpuTopology_Host* host_location) in TpuHostLocationExternal() function
45 int32 TpuHostLocationExternal::Id() const { in Id()49 std::vector<TpuCoreLocationExternal> TpuHostLocationExternal::Cores( in Cores()
126 const tensorflow::tpu::TpuHostLocationExternal TpuPlatform::GetTpuHostLocation() in GetTpuHostLocation()128 return tpu::TpuHostLocationExternal( in GetTpuHostLocation()
58 virtual const TpuHostLocationExternal GetTpuHostLocation() const = 0;
66 const tensorflow::tpu::TpuHostLocationExternal GetTpuHostLocation()
42 TpuHostLocationExternal host_location, int replica_count,
57 TpuComputationPlacer::AssignLocalDevices(TpuHostLocationExternal host_location, in AssignLocalDevices()
104 tf_tpu::TpuHostLocationExternal host = platform->GetTpuHostLocation(); in GetDefaultDeviceAssignment()