- Sort Score
- Num 10 results
- Language All
Results 1 - 2 of 2 for DeleteTrace (0.07 seconds)
-
tensorflow/c/eager/gradients.cc
tensor_ids[i] = ToId(tensors[i]); tensor_dtypes[i] = tensors[i]->DataType(); } return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes); } void Tape::DeleteTrace(const AbstractTensorHandle* t) { GradientTape::DeleteTrace(ToId(t)); } std::vector<int64_t> MakeTensorIDList( absl::Span<AbstractTensorHandle* const> tensors) { std::vector<int64_t> ids(tensors.size());
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Tue Feb 24 06:18:31 GMT 2026 - 19.6K bytes - Click Count (0) -
tensorflow/c/eager/gradients.h
// 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*); // Consumes the internal state of the tape (so cannot be called more than // once unless the tape is persistent) and produces the gradient of the target
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 12 05:11:17 GMT 2024 - 6.9K bytes - Click Count (0)