diff options
Diffstat (limited to 'candle-core/src/op.rs')
| -rw-r--r-- | candle-core/src/op.rs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-core/src/op.rs b/candle-core/src/op.rs index b18f868d..3fe52ebc 100644 --- a/candle-core/src/op.rs +++ b/candle-core/src/op.rs @@ -81,6 +81,7 @@ pub enum Op { kernel: Tensor, padding: usize, stride: usize, + dilation: usize, }, #[allow(dead_code)] @@ -89,6 +90,7 @@ pub enum Op { kernel: Tensor, padding: usize, stride: usize, + dilation: usize, }, #[allow(dead_code)] @@ -98,6 +100,7 @@ pub enum Op { padding: usize, output_padding: usize, stride: usize, + dilation: usize, }, AvgPool2D { |