- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for RegisterNotDifferentiable (0.28 sec)
-
tensorflow/c/eager/gradients_test.cc
Status s = StatusFromTF_Status(status.get()); CHECK_EQ(errors::OK, s.code()) << s.message(); } }; Status RegisterGradients(GradientRegistry* registry) { TF_RETURN_IF_ERROR(RegisterNotDifferentiable(registry, "CheckNumerics")); return absl::OkStatus(); } TEST_P(CppGradients, TestSetAttrString) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/not_differentiable.cc
absl::Span<AbstractTensorHandle*> grad_inputs) { for (int i = 0; i < grad_inputs.size(); i++) { grad_inputs[i] = nullptr; } return absl::OkStatus(); } Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) { return registry->Register(op, [](const ForwardOperation& op) { return new NotDifferentiableGradientFunction; }); } } // namespace gradients
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
return errors::InvalidArgument( "Provided null gradient_function for '", op_type, "'.\n", "If the intent is to treat this op as non-differentiable consider " "using RegisterNotDifferentiable or " "NotDifferentiableGradientFunction."); } return gradient_function->Compute(ctx_, output_gradients, result); } Status TapeVSpace::BuildOnesLike(const TapeTensor& t,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0)