- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AddRegisterer (0.17 sec)
-
tensorflow/c/eager/gradients.h
// ~AddGradientFunction() override {} // }; // // GradientFunction* AddRegisterer(const ForwardOperation& op) { // // More complex gradient functions can use inputs/attrs etc. from the // // forward `op`. // return new AddGradientFunction; // } // // Status RegisterGradients(GradientRegistry* registry) { // return registry->Register("Add", AddRegisterer); // } class GradientFunction { public:
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad_test.cc
y.reset(y_raw); } // TODO(srbs): Rename ops::Add to ops::AddV2 and AddRegister to // AddV2Registerer. status_ = registry_.Register("AddV2", AddRegisterer); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); ASSERT_NO_FATAL_FAILURE(CompareNumericalAndAutodiffGradients( AddModel, BuildGradModel(AddModel, registry_),
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.cc
// TODO(b/174778737): Only hold needed inputs and outputs. vector<AbstractTensorHandle*> forward_inputs_; vector<AbstractTensorHandle*> forward_outputs_; }; } // namespace GradientFunction* AddRegisterer(const ForwardOperation& op) { return new AddGradientFunction; } GradientFunction* ExpRegisterer(const ForwardOperation& op) { return new ExpGradientFunction(op.outputs[0]); }
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0)