Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_VariableInfo (0.43 sec)

  1. tensorflow/c/experimental/next_pluggable_device/c_api.cc

    struct TF_VariableInfo {
      TF_VariableInfo() = delete;
      // TF_VariableInfo is constructed here by TensorFlow, and will be passed to
      // plugin as a opaque pointer. Plugin will need to call C APIs below to
      // operate on TF_VariableInfo (such as allocate temp tensor for the `var` held
      // by the underlying tensorflow::VariableInfo.
      TF_VariableInfo(int index, const std::string& name, tensorflow::Var* var) {
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jan 09 00:52:04 GMT 2024
    - 13.9K bytes
    - Viewed (1)
  2. tensorflow/c/experimental/next_pluggable_device/c_api.h

    TF_CAPI_EXPORT extern void TF_AllocateTempForVariableInfo(
        TF_OpKernelContext* ctx, TF_VariableInfo* var_info, TF_Status* status);
    
    TF_CAPI_EXPORT extern TF_Tensor* TF_GetTensorFromVariableInfo(
        TF_VariableInfo* var_info, TF_Status* status);
    
    TF_CAPI_EXPORT extern void TF_DeleteVariableInfo(TF_VariableInfo* var_info);
    
    // ---------------------  Coordination service  --------------------------------
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Dec 20 20:01:06 GMT 2023
    - 7.2K bytes
    - Viewed (0)
Back to top