Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_ResetPjRtCClient (0.34 sec)

  1. 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)
  2. 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)
Back to top