- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_ResetPjRtCClient (0.27 sec)
-
tensorflow/c/experimental/next_pluggable_device/c_api.h
int num_options); // Resets the PjRt client for a device. After this, `TF_GetPjRtCClient` will // returns an error for that device. TF_CAPI_EXPORT extern void TF_ResetPjRtCClient(const char* device_type, TF_Status* status); // Gets the `PJRT_Client*` stored in TF global ResourceManager.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 20:01:06 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
return; } absl::Status s = tensorflow::SetPjRtClientInTFGlobalResourceManager( 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)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 13.9K bytes - Viewed (0)