Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h167 std::map<int, std::shared_ptr<PjRtDevice>> id_to_device_; variable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.h249 absl::flat_hash_map<int, TfrtCpuDevice*> id_to_device_; variable
Dpjrt_stream_executor_client.h357 std::map<int, PjRtDevice*> id_to_device_; variable