Home
last modified time | relevance | path

Searched refs:num_tpus_per_host (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.cc152 int* num_tpus_per_host, std::vector<std::vector<Device*>>* tpu_devices) { in GetTPUDevices() argument
166 *num_tpus_per_host = 0; in GetTPUDevices()
186 *num_tpus_per_host = host_tpu_devices.size(); in GetTPUDevices()
187 } else if (*num_tpus_per_host != host_tpu_devices.size()) { in GetTPUDevices()
191 "Mismatched number of TPU devices in cluster ", *num_tpus_per_host, in GetTPUDevices()
Ddistributed_tpu_rewrite_helpers.h63 int* num_tpus_per_host,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc137 int num_tpus_per_host = 0; in GetTPUDevices() local
141 num_tpus_per_host = host_tpu_devices.size(); in GetTPUDevices()
150 if (num_tpus_per_host != host_tpu_devices_size) in GetTPUDevices()
153 num_tpus_per_host, ", got ", host_tpu_devices.size()); in GetTPUDevices()