Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TapeTensor (0.16 sec)

  1. tensorflow/c/eager/tape.h

    void GradientTape<Gradient, BackwardFunction, TapeTensor>::Watch(
        int64_t tensor_id) {
      tensor_tape_.emplace(tensor_id, -1);
    }
    
    template <typename Gradient, typename BackwardFunction, typename TapeTensor>
    void GradientTape<Gradient, BackwardFunction, TapeTensor>::RecordOperation(
        const string& op_type, const std::vector<TapeTensor>& output_tensors,
    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.cc

      return absl::OkStatus();
    }
    
    TapeTensor::TapeTensor(AbstractTensorHandle* handle) : handle_(handle) {
      handle_->Ref();
    }
    TapeTensor::TapeTensor(const TapeTensor& other) {
      handle_ = other.handle_;
      handle_->Ref();
    }
    TapeTensor::~TapeTensor() { handle_->Unref(); }
    
    int64_t TapeTensor::GetID() const { return ToId(handle_); }
    
    tensorflow::DataType TapeTensor::GetDType() const {
      return handle_->DataType();
    }
    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)
Back to top