Home
last modified time | relevance | path

Searched defs:max_inflight_computations (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.cc66 int max_inflight_computations) in TpuDeviceState()
249 int max_inflight_computations, absl::Duration init_retry_timeout) { in GetTpuClient()
H A Dlocal_device_state.cc32 int max_inflight_computations, in LocalDeviceState()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla.cc335 [](int max_inflight_computations) -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()