Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for RecordOperation (0.23 sec)

  1. tensorflow/c/experimental/gradients/custom_gradient_test.cc

      AbstractTensorHandle* exp_output;
      TF_RETURN_IF_ERROR(ops::Exp(ctx, inputs[0], &exp_output, "Exp"));
      std::unique_ptr<GradientFunction> gradient_function(
          new PassThroughGradientFunction);
      tape.RecordOperation(inputs, {exp_output}, gradient_function.release());
      TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx,
                                              /*targets*/ {exp_output},
                                              /*sources=*/inputs,
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 4.8K bytes
    - Viewed (0)
  2. 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.
      Status CallBackwardFunction(
          const string& op_type, GradientFunction* gradient_function,
          const std::vector<int64_t>& unneeded_gradients,
          gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K 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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
Back to top