Searched defs:num_tpus_per_task (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 191 const int num_tpus_per_task = tpu_devices[0].size(); in GetFullMeshTPUExecutionDeviceAssignment() local
265 llvm::StringRef topology_attr, int num_tasks, int num_tpus_per_task) { in ParseTopologyAttr()
352 const int num_tpus_per_task = tpu_devices[0].size(); in GetGeneralTPUExecutionDeviceAssignment() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | topology.py | 221 def num_tpus_per_task(self): member in Topology
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1359 string* tpu_compilation_device, int* num_tpus_per_task, in GetTPUDeviceNames()
1386 int num_tasks, int num_tpus_per_task, in ParseTopologyAttr()
1504 int num_tasks, int num_tpus_per_task, in BuildFullMeshDeviceAssignment()
1562 const tpu::TpuTopologyExternal& tpu_topology, int num_tpus_per_task, in BuildDeviceAssignment()
3933 int num_tpus_per_task; in GetDeviceTopology() local
|