gpu test

FluxML · Apr 7, 2019 · 7cc3006 · 7cc3006
1 parent a4e2d0e
commit 7cc3006
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git 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
@@ -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))...)