diff --git a/src/Flux.jl b/src/Flux.jl index 994f6585..d3537b9e 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -8,7 +8,7 @@ using MacroTools: @forward @reexport using NNlib using Zygote: Params, @adjoint, gradient -export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, ConvTranspose, MaxPool, MeanPool, +export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, CrossCor, ConvTranspose, MaxPool, MeanPool, DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm, SkipConnection,params, mapleaves, cpu, gpu, f32, f64, param, data