Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for gradient_func (0.1 sec)

  1. 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;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

      // and string consistent.
      forward_op_->attrs.BuildNodeDef();
      std::unique_ptr<GradientFunction> gradient_fn;
      TF_RETURN_IF_ERROR(registry.Lookup(*forward_op_, &gradient_fn));
      tape->RecordOperation(forward_op_->inputs, retvals, gradient_fn.release(),
                            op_->Name());
      return absl::OkStatus();
    }
    }  // namespace internal
    
    }  // namespace gradients
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top