Add grads for interpolate1d (#1742)

* add backprop for interpolate1d * fix clippy lint * correct fix clippy lint
2025-06-16 10:38:54 +00:00 · 2024-02-22 07:44:01 +00:00
parent 45d5322d62
commit 8013b50829
4 changed files with 51 additions and 6 deletions
--- a/candle-core/src/op.rs
+++ b/candle-core/src/op.rs
@ -132,7 +132,10 @@ pub enum Op {
        stride: (usize, usize),
    },

-    UpsampleNearest1D(Tensor),
+    UpsampleNearest1D {
+        arg: Tensor,
+        target_size: usize,
+    },
    UpsampleNearest2D {
        arg: Tensor,
        target_h: usize,