tensorOp

Tensor operation : C = op( alpha1 * A, alpha2 * B ) + beta * C

- list of ops CUDNN_OP_TENSOR_ADD = 0, CUDNN_OP_TENSOR_MUL = 1, CUDNN_OP_TENSOR_MIN = 2, CUDNN_OP_TENSOR_MAX = 3, CUDNN_OP_TENSOR_SQRT = 4, CUDNN_OP_TENSOR_NOT = 5,

B tensor is ignored for CUDNN_OP_TENSOR_SQRT, CUDNN_OP_TENSOR_NOT.

version(grain_cuda)

void

tensorOp

(

cudnnOpTensorOp_t op

size_t dim

)

(

Variable!(T, dim, DeviceStorage) c

Variable!(T, dim, DeviceStorage) a

Variable!(T, dim, DeviceStorage) b

T alpha1 = 1

T alpha2 = 1

T beta = 0

)

tensorOp

Meta

Source