Searched defs:tpu_devices (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.h | 75 TPUDevicesAndHosts tpu_devices; member
|
| D | tpu_rewrite_device_util.cc | 123 llvm::SmallVectorImpl<llvm::SmallVector<Device, 8>>* tpu_devices) { in GetTPUDevices()
189 llvm::ArrayRef<llvm::SmallVector<Device, 8>> tpu_devices) { in GetFullMeshTPUExecutionDeviceAssignment()
348 llvm::ArrayRef<llvm::SmallVector<Device, 8>> tpu_devices, in GetGeneralTPUExecutionDeviceAssignment()
458 llvm::SmallVector<llvm::SmallVector<Device, 8>, 8> tpu_devices; in GetTPUCompilationAndExecutionDevices() local
|
| D | tpu_rewrite_device_util_test.cc | 333 const auto& tpu_devices = tpu_device_assignment.tpu_devices; in TEST() local
436 const auto& tpu_devices = tpu_device_assignment.tpu_devices; in TEST() local
550 auto& tpu_devices = tpu_device_assignment.tpu_devices; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 414 tpu_devices, in AssignDevicesToReplicate()
470 tpu_devices, in BuildParallelExecuteOp()
545 tpu_devices, in AssignDevicesToReplicatedExecute()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.cc | 152 int* num_tpus_per_host, std::vector<std::vector<Device*>>* tpu_devices) { in GetTPUDevices()
|
| D | distributed_tpu_rewrite_pass.cc | 1360 std::vector<std::vector<Device*>>* tpu_devices) { in GetTPUDeviceNames()
1503 int num_replicas, const std::vector<std::vector<Device*>>& tpu_devices, in BuildFullMeshDeviceAssignment()
1525 const std::vector<std::vector<Device*>>& tpu_devices, in BuildGeneralDeviceAssignment()
1563 const std::vector<std::vector<Device*>>& tpu_devices, int num_replicas, in BuildDeviceAssignment()
3932 std::vector<std::vector<Device*>> tpu_devices; in GetDeviceTopology() local
|