Searched defs:tpu_topology (Results 1 – 2 of 2) sorted by relevance
374 auto tpu_topology = ctx->input(0).scalar<tstring>()(); in Compute() local
1385 const tpu::TpuTopologyExternal& tpu_topology, in ParseTopologyAttr()1443 const tpu::TpuTopologyExternal& tpu_topology, int num_replicas, in ParseDeviceAssignmentAttr()1562 const tpu::TpuTopologyExternal& tpu_topology, int num_tpus_per_task, in BuildDeviceAssignment()3960 tpu::TpuTopologyExternal tpu_topology(tpu_platform->GetTopologyPtr()); in GetDeviceTopology() local