TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Revert 1 change

Browse Source
This commit is contained in:
Avik Pal 2018-06-28 14:25:22 +05:30
parent 5ccde88ce6
commit 7ac9e191cb
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/cuda/cuda.jl
View File

@ -3,7 +3,6 @@ module CUDA
using CuArrays
if CuArrays.cudnn_available()
CuParam{T,N} = Union{CuArray{T,N},TrackedArray{T,N,CuArray{T,N}}}
include("curnn.jl")
include("cudnn.jl")
end

1
src/cuda/curnn.jl
View File

@ -234,6 +234,7 @@ function copy_transpose!(dst::CuArray, src::CuArray)
return dst
end
CuParam{T,N} = Union{CuArray{T,N},TrackedArray{T,N,CuArray{T,N}}}
CuRNN{T} = Flux.RNNCell{<:Union{typeof(tanh),typeof(relu)},<:CuParam{T,2},<:CuParam{T,1}}
CuGRU{T} = Flux.GRUCell{<:CuParam{T,2},<:CuParam{T,1}}
CuLSTM{T} = Flux.LSTMCell{<:CuParam{T,2},<:CuParam{T,1}}