Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for gradient_func (0.56 sec)

  1. tensorflow/c/c_test_util.h

    bool GetAttrValue(TF_Operation* oper, const char* attr_name,
                      tensorflow::AttrValue* attr_value, TF_Status* s);
    
    // Returns a sorted vector of std::pair<function_name, gradient_func> from
    // graph_def.library().gradient()
    std::vector<std::pair<string, string>> GetGradDefs(
        const tensorflow::GraphDef& graph_def);
    
    // Returns a sorted vector of names contained in `grad_def`
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 09 01:06:53 GMT 2018
    - 6K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_function.cc

      if (TF_GetCode(status) != TF_OK) return;
    
      tensorflow::GradientDef gdef;
      gdef.set_function_name(func->record->fdef().signature().name());
      gdef.set_gradient_func(grad->record->fdef().signature().name());
      status->status = g->graph.AddGradientDef(std::move(gdef));
    }
    
    int TF_GraphNumFunctions(TF_Graph* g) {
      tensorflow::mutex_lock l(g->mu);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  3. tensorflow/c/c_test_util.cc

        const tensorflow::GraphDef& graph_def) {
      std::vector<std::pair<string, string>> grads;
      for (const tensorflow::GradientDef& grad : graph_def.library().gradient()) {
        grads.emplace_back(grad.function_name(), grad.gradient_func());
      }
      std::sort(grads.begin(), grads.end());
      return grads;
    }
    
    std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) {
      std::vector<string> names;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
Back to top