Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for RegisterNotDifferentiable (0.29 sec)

  1. 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)
  2. 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)
  3. tensorflow/c/experimental/gradients/not_differentiable.h

                     absl::Span<AbstractTensorHandle*> grad_inputs) override;
    };
    // Shorthand for registry->Register(op, new NotDifferentiableGradientFunction)
    Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op);
    }  // namespace gradients
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Dec 03 22:28:48 UTC 2020
    - 1.5K bytes
    - Viewed (0)
  4. 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)
Back to top