Home
last modified time | relevance | path

Searched refs:enumerate_devices_timeout (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla.cc484 std::optional<int> enumerate_devices_timeout, in PYBIND11_MODULE()
495 if (enumerate_devices_timeout.has_value()) { in PYBIND11_MODULE()
496 options.enumerate_devices_timeout = in PYBIND11_MODULE()
497 absl::Seconds(*enumerate_devices_timeout); in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Dservice.cc67 absl::ToInt64Milliseconds(options.enumerate_devices_timeout)); in EnableCoordinationService()
281 options_.enumerate_devices_timeout)) { in EnumerateDevices()
284 absl::FormatDuration(options_.enumerate_devices_timeout), in EnumerateDevices()
H A Dservice.h58 absl::Duration enumerate_devices_timeout = absl::Seconds(60); member
H A Dclient_server_test.cc480 service_options.enumerate_devices_timeout = timeout; in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A D__init__.pyi533 enumerate_devices_timeout: Optional[int] = ...,