- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetID (0.12 sec)
-
tensorflow/c/eager/gradients.h
}; // TODO(srbs): Figure out if we can avoid declaring this in the public header. // Wrapper for a tensor output of an operation executing under a tape. // // `GetID` returns a unique id for the wrapped tensor which is used to maintain // a map (`tensorflow::eager::TensorTape`) from the wrapped tensor to the id of // the op that produced it (or -1 if this tensor was watched using
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) -
tensorflow/c/eager/tape.h
for (const TapeTensor& o : output_tensors) { // Note: the tensor can have already been watched and hence be in the tape, // so we cannot check that we're inserting it here. tensor_tape_[o.GetID()] = op_id; tensor_usage_[o.GetID()] = 1; tensors.push_back(o); } op_tape_[op_id] = OpTapeEntry<BackwardFunction, TapeTensor>{ op_type, std::move(tensors), std::move(ids), backward_function_getter(),
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
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(); } AbstractTensorHandle* TapeTensor::GetHandle() const { return handle_; }
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)