- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ResetPjRtClient (0.57 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)