Searched defs:pjrt_client (Results 1 – 2 of 2) sorted by relevance
33 PyClient::PyClient(std::unique_ptr<PjRtClient> pjrt_client) in PyClient()35 PyClient::PyClient(std::shared_ptr<PjRtClient> pjrt_client) in PyClient()
94 PjRtClient* pjrt_client() const { return pjrt_client_.get(); } in pjrt_client() function