Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for RecordOperation (0.29 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)
  4. tensorflow/c/experimental/gradients/tape/tape_operation.cc

      forward_op_.attrs.BuildNodeDef();
      // TODO(b/170307493): Populate skip_input_indices here.
      std::unique_ptr<GradientFunction> backward_fn;
      TF_RETURN_IF_ERROR(registry_.Lookup(forward_op_, &backward_fn));
      tape_->RecordOperation(forward_op_.inputs, forward_op_.outputs,
                             backward_fn.release(), parent_op_->Name());
      return OkStatus();
    }
    
    }  // namespace gradients
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
Back to top