Searched refs:TpuTopologyExternal (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_topology.cc | 64 int32 TpuTopologyExternal::LogicalDevicesPerHost( in LogicalDevicesPerHost()
70 int32 TpuTopologyExternal::LogicalDevicesPerChip( in LogicalDevicesPerChip()
76 int32 TpuTopologyExternal::HostCount() const { in HostCount()
80 int32 TpuTopologyExternal::ChipsPerHost() const { in ChipsPerHost()
84 TpuTopologyChipBoundsExternal TpuTopologyExternal::chip_bounds() const { in chip_bounds()
90 bool TpuTopologyExternal::HasChip(int x, int y, int z) const { in HasChip()
94 TpuCoreLocationExternal TpuTopologyExternal::CoreForId( in CoreForId()
100 TpuCoreLocationExternal TpuTopologyExternal::Core(TpuCoreTypeEnum core_type, in Core()
107 std::vector<TpuCoreLocationExternal> TpuTopologyExternal::cores( in cores()
122 int TpuTopologyExternal::IdForHost(TpuDimensionsExternal host) const { in IdForHost()
[all …]
|
| D | tpu_platform_interface.h | 62 TpuTopologyExternal topology() { in topology()
63 return TpuTopologyExternal(GetTopologyPtr()); in topology()
|
| D | tpu_topology.h | 68 class TpuTopologyExternal {
70 explicit TpuTopologyExternal(SE_TpuTopology* topology) in TpuTopologyExternal() function
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tpu_client.cc | 140 tf_tpu::TpuTopologyExternal topology = in GetTpuDevices()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 266 const tpu::TpuTopologyExternal& topology, int num_tpus_per_task,
|
| D | distributed_tpu_rewrite_pass.cc | 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()
|