Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for RecordOperation (0.14 sec)

  1. tensorflow/c/eager/gradients.cc

      AbstractTensorHandle* AggregateGradients(
          gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override;
    
      // Calls the passed-in backward function.
      // op_type is the op's name provided in RecordOperation.
      absl::Status CallBackwardFunction(
          const string& op_type, GradientFunction* gradient_function,
          const std::vector<int64_t>& unneeded_gradients,
          gtl::ArraySlice<AbstractTensorHandle*> 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)
  2. tensorflow/c/eager/gradients_internal.h

    absl::Status SetAttrFunctionList(AbstractOperation*, const char* attr_name,
                                     absl::Span<const AbstractOperation*> values,
                                     ForwardOperation*);
    
    // Make the call to `Tape::RecordOperation`.
    absl::Status Execute(AbstractOperation*, AbstractContext*,
                         absl::Span<AbstractTensorHandle*> retvals,
                         int* num_retvals, ForwardOperation*, Tape*,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients_test.cc

      Tape tape(/*persistent=*/false);
      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,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/gradients.h

      // on the tape and marks all its outputs as watched if at
      // least one input of the op is watched and has a trainable dtype.
      // op_name is optional and is used for debugging only.
      void RecordOperation(absl::Span<AbstractTensorHandle* const> inputs,
                           absl::Span<AbstractTensorHandle* const> outputs,
                           GradientFunction* gradient_function,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.9K bytes
    - Viewed (0)
Back to top