Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ComputeGradient (0.14 sec)

  1. tensorflow/c/eager/gradients.h

                         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;
    
      // Adds this tensor to the list of watched tensors.
      //
    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_test.cc

      tape.Watch(inputs[0]);
      AbstractTensorHandle* neg_output;
      TF_RETURN_IF_ERROR(ops::Neg(ctx, inputs[0], &neg_output, "Neg"));
      tape.RecordOperation(inputs, {neg_output}, nullptr, "Neg");
      return tape.ComputeGradient(ctx,
                                  /*targets=*/{neg_output},
                                  /*sources=*/inputs,
                                  /*output_gradients=*/{}, outputs);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

      std::vector<int64_t> ids(tensors.size());
      for (int i = 0; i < tensors.size(); i++) {
        ids[i] = ToId(tensors[i]);
      }
      return ids;
    }
    
    absl::Status Tape::ComputeGradient(
        AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> targets,
        absl::Span<AbstractTensorHandle* const> sources,
        absl::Span<AbstractTensorHandle* const> output_gradients,
    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