From 62fc9656173029dbb3a14dc1938a819501242809 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Sun, 22 Oct 2023 06:57:03 +0100 Subject: [PATCH] Expose the track-op method. (#1148) --- candle-core/src/tensor.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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() }