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

cufunc

This commit is contained in:
cossio 2020-05-07 09:54:05 +02:00
parent 8314200c51
commit 86d6555269
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/layers/stateless.jl
View File

@ -271,6 +271,10 @@ function xlogx(x)
result = x * log(x)
ifelse(x > zero(x), result, zero(result))
end
CuArrays.@cufunc function xlogx(x)
result = x * log(x)
ifelse(x > zero(x), result, zero(result))
end
"""
xlogy(x, y)
@ -280,3 +284,7 @@ function xlogy(x, y)
result = x * log(y)
ifelse(x > zero(x), result, zero(result))
end
CuArrays.@cufunc function xlogy(x, y)
result = x * log(y)
ifelse(x > zero(x), result, zero(result))
end