Flux.jl/test/cuda
bors[bot] 472e1fbf5e
Merge #957
957: Add some gradient checking tests on GPUs r=dhairyagandhi96 a=dhairyagandhi96

Good to add generic tests for tracking gradients through the various layers on the GPU.

Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com>
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
2020-05-21 12:25:53 +00:00
..
cuda.jl Merge #957 2020-05-21 12:25:53 +00:00
cudnn.jl
curnn.jl
layers.jl