Searched defs:num_devices_per_host (Results 1 – 2 of 2) sorted by relevance
102 std::vector<int32_t> num_devices_per_host; in ConstructDevicesPerHost() local129 xla::StatusOr<std::vector<int32_t>> num_devices_per_host = in Compute() local165 size_t num_devices_per_host = -1; in Compute() local
146 ResourceMgr* rmgr, const std::vector<int32_t>& num_devices_per_host, in ConstructTpuPodState()