diff options
Diffstat (limited to 'candle-nn/src/lib.rs')
| -rw-r--r-- | candle-nn/src/lib.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-nn/src/lib.rs b/candle-nn/src/lib.rs index 46a83800..ae955f56 100644 --- a/candle-nn/src/lib.rs +++ b/candle-nn/src/lib.rs @@ -2,8 +2,8 @@ // error type if needed or add some specialized cases on the candle-core side. pub mod activation; pub mod conv; -pub mod dataset; pub mod embedding; +pub mod group_norm; pub mod init; pub mod layer_norm; pub mod linear; @@ -11,11 +11,11 @@ pub mod loss; pub mod ops; pub mod optim; pub mod var_builder; -pub mod vision; pub use activation::Activation; -pub use conv::{Conv1d, Conv1dConfig}; +pub use conv::{conv1d, conv2d, Conv1d, Conv1dConfig, Conv2d, Conv2dConfig}; pub use embedding::{embedding, Embedding}; +pub use group_norm::{group_norm, GroupNorm}; pub use init::Init; pub use layer_norm::{layer_norm, LayerNorm}; pub use linear::{linear, linear_no_bias, Linear}; |