diff --git a/src/Flux.jl b/src/Flux.jl index b117aa4b9c..a041a69a8d 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -6,9 +6,9 @@ using Base: tail using MacroTools, Juno, Requires, Reexport, Statistics, Random using MacroTools: @forward -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, - params, mapleaves, cpu, gpu, f32, f64, CrossCor + params, mapleaves, cpu, gpu, f32, f64 @reexport using NNlib diff --git a/src/layers/conv.jl b/src/layers/conv.jl index 68ce42928a..76923f3f0b 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -192,11 +192,6 @@ struct CrossCor{N,F,A,V} dilation::NTuple{N,Int} end -function crosscor(x, w, d) - - return conv(x, w, DenseConvDims(d)) -end - CrossCor(w::AbstractArray{T,N}, b::AbstractVector{T}, σ = identity; stride = 1, pad = 0, dilation = 1) where {T,N} = CrossCor(σ, w, b, expand.(sub2(Val(N)), (stride, pad, dilation))...)