- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for grad_outputs (0.28 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
output->Ref(); } } } Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { AbstractTensorHandle* upstream_grad = grad_outputs[0]; AbstractTensorHandle* activations = forward_outputs_[0]; // Calculate Grad std::string name = "relu_grad";
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/math_grad.cc
public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { // TODO(b/161805092): Support broadcasting. DCHECK(grad_outputs[0]); grad_inputs[0] = grad_outputs[0]; grad_inputs[1] = grad_outputs[0]; grad_inputs[0]->Ref(); grad_inputs[1]->Ref(); return absl::OkStatus(); }
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/experimental/gradients/array_grad.cc
public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { for (int i = 0; i < grad_outputs.size(); i++) { auto grad_input = grad_outputs[i]; // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle // that takes care of this similar to `Tensor`?
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/c_api_test.cc
void RunGraphsAndCompareOutputs(TF_Output* grad_outputs, TF_Output* expected_grad_outputs) { std::unique_ptr<CSession> csession(new CSession(graph_, s_)); std::unique_ptr<CSession> expected_csession( new CSession(expected_graph_, s_)); std::vector<TF_Output> grad_outputs_vec; grad_outputs_vec.assign(grad_outputs, grad_outputs + 2); csession->SetOutputs(grad_outputs_vec);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
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/eager/gradients.h
// absl::Span<AbstractTensorHandle* const> grad_inputs, // absl::Span<AbstractTensorHandle*> grad_outputs) override { // grad_outputs[0] = grad_inputs[0]; // grad_outputs[1] = grad_inputs[0]; // grad_outputs[0]->Ref(); // grad_outputs[1]->Ref(); // return OkStatus(); // } // ~AddGradientFunction() override {} // }; //
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/custom_gradient_test.cc
public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { CHECK_EQ(grad_outputs.size(), 1); CHECK_EQ(grad_inputs.size(), 1); grad_inputs[0] = grad_outputs[0]; if (grad_inputs[0]) { grad_inputs[0]->Ref(); } return absl::OkStatus(); } };
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/not_differentiable.cc
#include "tensorflow/c/experimental/gradients/not_differentiable.h" namespace tensorflow { namespace gradients { Status NotDifferentiableGradientFunction::Compute( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) { for (int i = 0; i < grad_inputs.size(); i++) { grad_inputs[i] = nullptr; } return OkStatus(); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 15 01:15:58 GMT 2022 - 1.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
n->name(), "). This is a bug. Please file an issue at " "https://github.com/tensorflow/tensorflow/issues."); } } } // Unpack the results from grad_outputs_arg. TFOutputsFromOutputs(dy_arg, dy); #endif // defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD) } // TF_Session functions ----------------------------------------------
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0)