- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TapeTensor (0.16 sec)
-
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) -
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) -
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;
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)