Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ShouldRecord (0.08 sec)

  1. tensorflow/c/eager/gradients.h

                           GradientFunction* gradient_function,
                           const string& op_name = "");
      // Returns whether any tensor in a list of tensors is being watched and has
      // a trainable dtype.
      bool ShouldRecord(
          absl::Span<const AbstractTensorHandle* const> tensors) const;
      // Unwatches this tensor on the tape. Mainly used for cleanup when deleting
      // eager tensors.
      void DeleteTrace(const AbstractTensorHandle*);
    
    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.cc

          [gradient_function]() -> GradientFunction* { return gradient_function; },
          [](GradientFunction* ptr) {
            if (ptr) {
              delete ptr;
            }
          });
    }
    bool Tape::ShouldRecord(
        absl::Span<const AbstractTensorHandle* const> tensors) const {
      std::vector<int64_t> tensor_ids(tensors.size());
      std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
    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