Searched defs:tpu_topology (Results 1 – 5 of 5) 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
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