Searched refs:host_location (Results 1 – 3 of 3) sorted by relevance
57 TpuComputationPlacer::AssignLocalDevices(TpuHostLocationExternal host_location, in AssignLocalDevices() argument63 host_location.impl(), replica_count, computation_count, result.data(), in AssignLocalDevices()
42 TpuHostLocationExternal host_location, int replica_count,
51 explicit TpuHostLocationExternal(SE_TpuTopology_Host* host_location) in TpuHostLocationExternal() argument52 : host_location_(host_location) {} in TpuHostLocationExternal()