huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose the track-op method. (#1148)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-22 06:57:03 +01:00
committed by GitHub
parent 5b32c2a41e
commit 62fc965617
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/tensor.rs
View File

@ -449,7 +449,7 @@ impl Tensor {
/// Returns true if the computation graph should track this op, that is if it is
/// a variable or if it has some variable as dependencies.
pub(crate) fn track_op(&self) -> bool {
pub fn track_op(&self) -> bool {
self.is_variable || self.op.is_some()
}