- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for Check (0.12 sec)
-
tensorflow/c/eager/tape.h
bool GradientTape<Gradient, BackwardFunction, TapeTensor>::ShouldRecord( absl::Span<const int64_t> tensor_ids, absl::Span<const tensorflow::DataType> dtypes) const { CHECK_EQ(tensor_ids.size(), dtypes.size()); for (int i = 0; i < tensor_ids.size(); ++i) { if (tensor_tape_.find(tensor_ids[i]) != tensor_tape_.end()) { if (IsDtypeTrainable(dtypes[i])) { return true; }
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)