Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. 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();
    }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.h

    // Figure out a way to avoid this.
    // TODO(srbs): Should ZerosLike check-fail instead of returning nullptr?
    class TapeTensor {
     public:
      explicit TapeTensor(AbstractTensorHandle* handle);
      TapeTensor(const TapeTensor& other);
      ~TapeTensor();
    
      int64_t GetID() const;
      tensorflow::DataType GetDType() const;
    
      AbstractTensorHandle* ZerosLike() const;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 6.9K bytes
    - Viewed (0)
Back to top