- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for gradient_function (0.16 sec)
-
tensorflow/c/experimental/gradients/custom_gradient_test.cc
tape.Watch(inputs[0]); // Watch x. AbstractTensorHandle* exp_output; TF_RETURN_IF_ERROR(ops::Exp(ctx, inputs[0], &exp_output, "Exp")); std::unique_ptr<GradientFunction> gradient_function( new PassThroughGradientFunction); tape.RecordOperation(inputs, {exp_output}, gradient_function.release()); TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, /*targets*/ {exp_output},
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
}; } // namespace GradientFunction* ReluRegisterer(const ForwardOperation& op) { return new ReluGradientFunction(op.outputs); } GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op) { return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs); } GradientFunction* BiasAddRegisterer(const ForwardOperation& op) {
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad.cc
#include "tensorflow/c/experimental/gradients/array_grad.h" #include "tensorflow/c/eager/abstract_context.h" namespace tensorflow { namespace gradients { namespace { class IdentityNGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override {
C++ - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 1.6K bytes - Viewed (0)