Home
last modified time | relevance | path

Searched refs:GetTpuClient (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.h54 StatusOr<std::shared_ptr<PjRtClient>> GetTpuClient(
Dtpu_client.cc175 StatusOr<std::shared_ptr<PjRtClient>> GetTpuClient( in GetTpuClient() function
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc261 GetTpuClient(asynchronous)); in PYBIND11_MODULE()