Searched refs:pjrt_c_api (Results 1 – 3 of 3) sorted by relevance
286 const PJRT_Api* PjRtCApiClient::pjrt_c_api() const { return c_api_; } in pjrt_c_api() function in xla::PjRtCApiClient303 const PJRT_Api* api = client_->pjrt_c_api(); in id()313 const PJRT_Api* api = client_->pjrt_c_api(); in process_index()323 const PJRT_Api* api = client_->pjrt_c_api(); in IsAddressable()334 const PJRT_Api* api = client_->pjrt_c_api(); in InitAttributes()373 const PJRT_Api* c_api = client_->pjrt_c_api(); in device_kind()385 const PJRT_Api* api = client_->pjrt_c_api(); in local_hardware_id()395 const PJRT_Api* c_api = client_->pjrt_c_api(); in DebugString()406 const PJRT_Api* c_api = client_->pjrt_c_api(); in ToString()436 const PJRT_Api* api = pjrt_c_api(); in InitDevices()[all …]
276 const PJRT_Api* pjrt_c_api() const;407 const PJRT_Api* pjrt_c_api() const { return client_->pjrt_c_api(); } in pjrt_c_api() function500 const PJRT_Api* pjrt_c_api() const { return client_->pjrt_c_api(); } in pjrt_c_api() function
5 hdrs = ["pjrt_c_api.h"],