diff options
Diffstat (limited to 'candle-nn/src/lib.rs')
| -rw-r--r-- | candle-nn/src/lib.rs | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/candle-nn/src/lib.rs b/candle-nn/src/lib.rs index 6e268f4e..8e5580df 100644 --- a/candle-nn/src/lib.rs +++ b/candle-nn/src/lib.rs @@ -16,7 +16,10 @@ pub mod var_map; pub use activation::Activation; pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig}; -pub use conv::{conv1d, conv2d, conv2d_no_bias, Conv1d, Conv1dConfig, Conv2d, Conv2dConfig}; +pub use conv::{ + conv1d, conv2d, conv2d_no_bias, conv_transpose2d, conv_transpose2d_no_bias, Conv1d, + Conv1dConfig, Conv2d, Conv2dConfig, ConvTranspose2d, ConvTranspose2dConfig, +}; pub use embedding::{embedding, Embedding}; pub use func::{func, Func}; pub use group_norm::{group_norm, GroupNorm}; |