Searched refs:num_tpu_devices (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_cloud_tpu_test.py | 72 for task in range(FLAGS.num_tpu_devices // DEVICES_PER_TASK):
|
| D | BUILD | 1113 args = ["--num_tpu_devices=32"],
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 192 const int num_tpu_devices = num_tasks * num_tpus_per_task; in GetFullMeshTPUExecutionDeviceAssignment() local
194 if (num_replicas != 1 && num_replicas != num_tpu_devices) in GetFullMeshTPUExecutionDeviceAssignment()
196 num_tpu_devices, ", got ", num_replicas); in GetFullMeshTPUExecutionDeviceAssignment()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1569 const int num_tpu_devices = num_tasks * num_tpus_per_task; in BuildDeviceAssignment() local
1574 if (num_replicas > num_tpu_devices) { in BuildDeviceAssignment()
1577 " but there are only ", num_tpu_devices, in BuildDeviceAssignment()
1581 " but there are only ", num_tpu_devices, in BuildDeviceAssignment()
1585 if (num_replicas * num_cores_per_replica > num_tpu_devices) { in BuildDeviceAssignment()
1589 num_tpu_devices, " cores in the TPU topology"); in BuildDeviceAssignment()
1605 if (num_replicas != 1 && num_replicas != num_tpu_devices) { in BuildDeviceAssignment()
1609 num_tpu_devices, in BuildDeviceAssignment()
1613 num_tpu_devices, ")"); in BuildDeviceAssignment()
|