Documentation
GitHub
Example
Dlang.org
poolForward
grain
cudnn
wrapper of cudnnPoolingForward for Variable
version
(
grain_cuda
)
auto
poolForward
(
bool
isMax
= true
bool
isAveragePad
= false
T
size_t
_tensorDims
size_t
_poolDims
)
(
Variable
!(
T
,
_tensorDims
,
DeviceStorage
)
input
,
int
[
_poolDims
]
windowDim
,
int
[
_poolDims
]
padding
,
int
[
_poolDims
]
stride
,
T
alpha
= 1
,
T
beta
= 0
)
Meta
Source
See Implementation
grain
cudnn
functions
activationBackward
activationForward
convBackward
convForward
cudnnDataType
fill
isContiguous
isDeterministic
isNanProp
makeCudnnTensor
poolBackward
poolForward
reduce
scale
softmaxBackward
softmaxForward
tensorOp
transform
structs
TensorDesc
ThreadLocalDropout
wrapper of cudnnPoolingForward for Variable