Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_LockVariableInfos (0.27 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/next_pluggable_device/c_api.h 

    TF_CAPI_EXPORT extern TF_VariableInfo* TF_CreateVariableInfoFromContext(
    TF_OpKernelContext* ctx, int index, TF_Status* status);

TF_CAPI_EXPORT extern void TF_LockVariableInfos(TF_VariableInfo** vars,
                                                int num_vars,
                                                TF_Status* status);

TF_CAPI_EXPORT extern void TF_AllocateTempForVariableInfo(
    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)
  3. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/next_pluggable_device/c_api.cc 

      tensorflow::Var* variable;
  cc_status = tensorflow::LookupResource(cc_ctx, handle, &variable);
  return new TF_VariableInfo(index, handle.name(), variable);
}

void TF_LockVariableInfos(TF_VariableInfo** vars, int num_vars,
                          TF_Status* status) {
  std::vector<tensorflow::VariableInfo*> variable_ptrs;
  variable_ptrs.reserve(num_vars);
  for (int i = 0; i < num_vars; ++i) {
    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)
Back to top