Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h140 std::map<int, std::shared_ptr<PjRtDevice>> id_to_device_; variable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h260 std::map<int, PjRtDevice*> id_to_device_; variable