- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for GradientFunction (0.22 sec)
-
tensorflow/c/experimental/gradients/math_grad.h
namespace tensorflow { namespace gradients { GradientFunction* AddRegisterer(const ForwardOperation& op); GradientFunction* ExpRegisterer(const ForwardOperation& op); GradientFunction* MatMulRegisterer(const ForwardOperation& op); GradientFunction* SqrtRegisterer(const ForwardOperation& op); GradientFunction* NegRegisterer(const ForwardOperation& op); GradientFunction* SubRegisterer(const ForwardOperation& op);
C - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.h
#include "tensorflow/c/eager/gradients.h" namespace tensorflow { namespace gradients { GradientFunction* ReluRegisterer(const ForwardOperation& op); GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer( const ForwardOperation& op); GradientFunction* BiasAddRegisterer(const ForwardOperation& op); } // namespace gradients } // namespace tensorflow
C - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1.2K 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/eager/gradients.h
GradientFunction, TapeTensor> { public: using GradientTape<AbstractTensorHandle, GradientFunction, TapeTensor>::GradientTape; // Returns whether the tape is persistent, i.e., whether the tape will hold // onto its internal state after a call to `ComputeGradient`. using GradientTape<AbstractTensorHandle, GradientFunction, TapeTensor>::IsPersistent;
C - Registered: Tue Apr 23 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.cc
} GradientFunction* SqrtRegisterer(const ForwardOperation& op) { return new SqrtGradientFunction(op.outputs[0]); } GradientFunction* NegRegisterer(const ForwardOperation& op) { return new NegGradientFunction; } GradientFunction* SubRegisterer(const ForwardOperation& op) { return new SubGradientFunction; } GradientFunction* MulRegisterer(const ForwardOperation& op) { return new MulGradientFunction(op.inputs);
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) -
tensorflow/c/eager/gradients.cc
tape_tensors.push_back(TapeTensor(t)); } GradientTape::RecordOperation( op_name, tape_tensors, input_ids, input_dtypes, [gradient_function]() -> GradientFunction* { return gradient_function; }, [](GradientFunction* ptr) { if (ptr) { delete ptr; } }); } bool Tape::ShouldRecord( absl::Span<const AbstractTensorHandle* const> tensors) const {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K 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) -
tensorflow/c/experimental/gradients/not_differentiable.h
#include "tensorflow/c/eager/gradients.h" namespace tensorflow { namespace gradients { // Ignores `grad_outputs` and sets all entries in grad_inputs to nullptr. class NotDifferentiableGradientFunction : public GradientFunction { Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override; };
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/custom_gradient_test.cc
Status s = StatusFromTF_Status(status.get()); CHECK_EQ(errors::OK, s.code()) << s.message(); } }; class PassThroughGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override {
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/array_grad.h
#ifndef TENSORFLOW_C_EXPERIMENTAL_GRADIENTS_ARRAY_GRAD_H_ #define TENSORFLOW_C_EXPERIMENTAL_GRADIENTS_ARRAY_GRAD_H_ #include "tensorflow/c/eager/gradients.h" namespace tensorflow { namespace gradients { GradientFunction* IdentityNRegisterer(const ForwardOperation& op); } // namespace gradients } // namespace tensorflow
C - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Thu Dec 03 22:28:48 GMT 2020 - 1K bytes - Viewed (0)