- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ShouldRecord (0.11 sec)
-
tensorflow/c/eager/gradients.h
GradientFunction* gradient_function, const string& op_name = ""); // Returns whether any tensor in a list of tensors is being watched and has // 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*);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
[gradient_function]() -> GradientFunction* { return gradient_function; }, [](GradientFunction* ptr) { if (ptr) { delete ptr; } }); } bool Tape::ShouldRecord( absl::Span<const AbstractTensorHandle* const> tensors) const { std::vector<int64_t> tensor_ids(tensors.size()); std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.7K bytes - Viewed (0)