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