Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GradientFunction (0.1 sec)

  1. 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;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  2. 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 {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top