Searched defs:tpu_topology (Results 1 – 9 of 9) sorted by relevance
140 tpu::TopologyProto* tpu_topology) { in InitializeTPUSystemGlobally()202 Status InitializeTPUSystemGlobally(Env* env, tpu::TopologyProto* tpu_topology) { in InitializeTPUSystemGlobally()214 tensorflow::tpu::TopologyProto tpu_topology; in InitializeTPUSystemGlobally() local
161 const auto tpu_topology = tpu_platform->topology(); in InitializeInternal() local281 auto tpu_topology = ctx->input(0).scalar<tstring>()(); in Compute() local
200 tpu::TpuTopologyExternal tpu_topology = tpu_platform->topology(); in TPUCoreIDsToLocations() local234 tpu::TpuTopologyExternal tpu_topology = tpu_platform->topology(); in TPUCoreLocationsToIDs() local
58 string tpu_topology = 4 [(google.api.field_behavior) = IMMUTABLE]; field
447 auto tpu_topology = ctx->input(0).scalar<tstring>()(); in Compute() local
1594 const tpu::TpuTopologyExternal& tpu_topology, in ParseTopologyAttr()1653 const tpu::TpuTopologyExternal& tpu_topology, int num_replicas, in ParseDeviceAssignmentAttr()1776 const tpu::TpuTopologyExternal& tpu_topology, int num_tpus_per_task, in BuildDeviceAssignment()4415 tpu::TpuTopologyExternal tpu_topology(tpu_platform->GetTopologyPtr()); in GetDeviceTopology() local
3508 string tpu_topology = 2 [(google.api.field_behavior) = OPTIONAL]; field
3989 string tpu_topology = 2; field