- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetTPUDevicesAndHostsNotReplicated (0.4 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
return num_cores_per_replica_attr.getInt(); } else { return 1; } } // Get the TPUDevicesAndHosts for a cluster that is not replicated. mlir::LogicalResult GetTPUDevicesAndHostsNotReplicated( mlir::TF::RuntimeDevices devices, mlir::tf_device::ClusterOp cluster, tensorflow::TPUDevicesAndHosts& devices_and_hosts) { std::string topology; if (failed(GetTopology(cluster, topology))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0)