grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
activationForward
grain
cudnn
y = alpha * f(x) + beta * y
version(grain_cuda)
void
activationForward
(
cudnnActivationMode_t
A
T
size_t
dim
)
(
Variable
!(
T
,
dim
,
DeviceStorage
)
x
,
Variable
!(
T
,
dim
,
DeviceStorage
)
y
,
T
alpha
= 1.0
,
T
beta
= 0.0
,
double
coeff
= 0.0
)
Meta
Source
See Implementation
grain
cudnn
functions
activationBackward
activationForward
contiguous
convBackward
convForward
cudnnDataType
fill
isContiguous
isDeterministic
isNanProp
makeCudnnTensor
poolBackward
poolForward
reduce
scale
softmaxBackward
softmaxForward
tensorOp
transform
static variables
deterministic
nanProp
structs
CudnnDropout
TensorDesc
ThreadLocalDropout
y = alpha * f(x) + beta * y