- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ResetPjRtClient (0.2 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type); absl::Status ResetPjRtClient(const DeviceType& device_type); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
device_type.type_string(), " is not type PjRtCApiClient")); } return pjrt_c_api_client; } absl::Status ResetPjRtClient(const DeviceType& device_type) { ResourceMgr* rmgr = tfrt_global::GetTFGlobalResourceMgr(); PjRtState* pjrt_state; TF_RETURN_IF_ERROR(rmgr->Lookup(rmgr->default_container(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
tensorflow::DeviceType(device_type), std::move(*pjrt_client)); status->status = s; } void TF_ResetPjRtCClient(const char* device_type, TF_Status* status) { status->status = tensorflow::ResetPjRtClient(tensorflow::DeviceType(device_type)); } PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) { absl::StatusOr<xla::PjRtCApiClient*> pjrt_c_api_client =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 13.9K bytes - Viewed (0)