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

rm CUDAnative mention

This commit is contained in:
Dhairya Gandhi 2020-06-01 22:31:23 +05:30
parent dd4a220da2
commit 39e6a70d1f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/cuda/curnn.jl
View File

@ -1,5 +1,4 @@
import ..Flux: Flux, relu
using CUDA.CUDAnative
CuRNN{T} = Flux.RNNCell{<:Union{typeof(tanh),typeof(relu)},<:CuArray{T,2},<:CuArray{T,1}}
CuGRU{T} = Flux.GRUCell{<:CuArray{T,2},<:CuArray{T,1}}