burn_backend/backend/ops/modules/
mod.rs

1/// Module with convolution operations.
2pub mod conv;
3
4/// Module with attention operations.
5pub mod attention;
6
7/// Module with unfold operations.
8pub mod unfold;
9
10/// Module with pooling operations.
11pub mod pool;
12
13/// Module for grid_sample operations
14pub mod grid_sample;
15
16mod base;
17
18pub use base::*;