- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for gradient_func (0.1 sec)
-
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) -
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)