Home
last modified time | relevance | path

Searched defs:num_devices_per_host (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_configuration_ops.cc102 std::vector<int32_t> num_devices_per_host; in ConstructDevicesPerHost() local
129 xla::StatusOr<std::vector<int32_t>> num_devices_per_host = in Compute() local
165 size_t num_devices_per_host = -1; in Compute() local
Dtpu_pod_state.cc146 ResourceMgr* rmgr, const std::vector<int32_t>& num_devices_per_host, in ConstructTpuPodState()