diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-07 17:15:38 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-07 16:15:38 +0100 |
commit | 2345b8ce3f8ebab6e04d6ea25f7c809efb037995 (patch) | |
tree | a1c74ed8d29d1f14d329eab6e1900749b041bbdd /candle-examples/examples/stable-diffusion/utils.rs | |
parent | f53a333ea91233b41dd946c2c30213c79b4d1cb3 (diff) | |
download | candle-2345b8ce3f8ebab6e04d6ea25f7c809efb037995.tar.gz candle-2345b8ce3f8ebab6e04d6ea25f7c809efb037995.tar.bz2 candle-2345b8ce3f8ebab6e04d6ea25f7c809efb037995.zip |
Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337)
* Skeleton for the avg-pool2d and upsample-nearest2d ops.
* Preliminary conv2d support.
Diffstat (limited to 'candle-examples/examples/stable-diffusion/utils.rs')
-rw-r--r-- | candle-examples/examples/stable-diffusion/utils.rs | 8 |
1 files changed, 0 insertions, 8 deletions
diff --git a/candle-examples/examples/stable-diffusion/utils.rs b/candle-examples/examples/stable-diffusion/utils.rs index 08b78c04..0c95cfef 100644 --- a/candle-examples/examples/stable-diffusion/utils.rs +++ b/candle-examples/examples/stable-diffusion/utils.rs @@ -1,13 +1,5 @@ use candle::{Device, Result, Tensor}; -pub fn avg_pool2d(_: &Tensor) -> Result<Tensor> { - todo!() -} - -pub fn upsample_nearest2d(_: &Tensor) -> Result<Tensor> { - todo!() -} - pub fn linspace(start: f64, stop: f64, steps: usize) -> Result<Tensor> { if steps < 1 { candle::bail!("cannot use linspace with steps {steps} <= 1") |