Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/eager/tape.h

      //
      // op_type is used to decide which of the incoming gradients can be left as
      // nullptr instead of building zeros when build_default_zeros_grads == true.
      void RecordOperation(
          const string& op_type, const std::vector<TapeTensor>& output_tensors,
          absl::Span<const int64_t> input_tensor_id,
          absl::Span<const tensorflow::DataType> input_dtypes,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradients_internal.h

    Status SetAttrFunctionList(AbstractOperation*, const char* attr_name,
                               absl::Span<const AbstractOperation*> values,
                               ForwardOperation*);
    
    // Make the call to `Tape::RecordOperation`.
    Status Execute(AbstractOperation*, AbstractContext*,
                   absl::Span<AbstractTensorHandle*> retvals, int* num_retvals,
                   ForwardOperation*, Tape*, const GradientRegistry&);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:35 GMT 2021
    - 4.2K bytes
    - Viewed (0)
  3. 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,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Sep 26 10:27:05 GMT 2022
    - 6.9K bytes
    - Viewed (0)
Back to top