diff options
| -rw-r--r-- | candle-core/src/tensor.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/tensor.rs b/candle-core/src/tensor.rs index 0ffed2fe..381c768b 100644 --- a/candle-core/src/tensor.rs +++ b/candle-core/src/tensor.rs @@ -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() } |