grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
to
grain
autograd
CUDA -> CPU memory conversion
auto
to
(T[] src)
version(grain_cuda)
auto
to
(
alias
S
:
DeviceStorage
T
)
(
T
[]
src
)
auto
to
(Src src)
Variable
!(
T
,
dim
,
Dst
)
to
(Variable!(T, dim, Src) src)
Meta
Source
See Implementation
grain
autograd
aliases
DeviceStorage
ElementType
HostStorage
isDevice
functions
gradSlice
iterVariables
length
refIterVariables
to
uninit
uninitVariable
variable
zero_
zeros
structs
BackProp
UntypedVariable
Variable
variables
DimensionCount
isHost
isVariable
CUDA -> CPU memory conversion