Dnnl

object Dnnl : ExternalLib

Functions

add

avgPool

external fun avgPool(resultShape: IntArray, result: FloatArray, imagesShape: IntArray, images: FloatArray, poolHeight: Int, poolWidth: Int)

avgPoolGrad

external fun avgPoolGrad(resultShape: IntArray, result: FloatArray, seedShape: IntArray, seed: FloatArray, poolHeight: Int, poolWidth: Int)

batchNorm

external fun batchNorm(resultShape: IntArray, result: FloatArray, mean: FloatArray, variance: FloatArray, input: FloatArray, scaleShift: FloatArray)

batchNormGrad

Convenience wrapper for DNNL batchnorm grad.

conv2d

external fun conv2d(resultShape: IntArray, result: FloatArray, inputShape: IntArray, input: FloatArray, filtersShape: IntArray, filters: FloatArray, hstride: Int, vstride: Int, paddingLeft: Int, paddingRight: Int, paddingTop: Int, paddingBottom: Int)

conv2dGradFilter

external fun conv2dGradFilter(resultShape: IntArray, result: FloatArray, seedShape: IntArray, seed: FloatArray, imagesShape: IntArray, images: FloatArray, hstride: Int, vstride: Int, paddingLeft: Int, paddingRight: Int, paddingTop: Int, paddingBottom: Int)

conv2dGradImage

external fun conv2dGradImage(resultShape: IntArray, result: FloatArray, seedShape: IntArray, seed: FloatArray, filtersShape: IntArray, filters: FloatArray, hstride: Int, vstride: Int, paddingLeft: Int, paddingRight: Int, paddingTop: Int, paddingBottom: Int)

linear

external fun linear(shape: IntArray, strides: IntArray, offset: Int, res: FloatArray, input: FloatArray, scale: Float, shift: Float)

logSoftmax

external fun logSoftmax(shape: IntArray, input: FloatArray, res: FloatArray, axis: Int)

logSoftmaxGrad

external fun logSoftmaxGrad(shape: IntArray, grad: FloatArray, seed: FloatArray, fwdRes: FloatArray, axis: Int)

Given the result of the forward op and the seed, returns the grad

matmul

external fun matmul(lhsShape: IntArray, lhsStrides: IntArray, lhsOffset: Int, rhsShape: IntArray, rhsStrides: IntArray, rhsOffset: Int, result: FloatArray, lhs: FloatArray, rhs: FloatArray)

maxPool

external fun maxPool(resultShape: IntArray, result: FloatArray, workspace: ByteArray, imagesShape: IntArray, images: FloatArray, poolHeight: Int, poolWidth: Int)

maxPoolGrad

external fun maxPoolGrad(resultShape: IntArray, result: FloatArray, workspace: ByteArray, seedShape: IntArray, seed: FloatArray, poolHeight: Int, poolWidth: Int)

mulScalar

reduceSum

external fun reduceSum(resultShape: IntArray, result: FloatArray, inputShape: IntArray, input: FloatArray)

relu

external fun relu(shape: IntArray, result: FloatArray, input: FloatArray)

reluGrad

external fun reluGrad(shape: IntArray, result: FloatArray, seed: FloatArray, input: FloatArray)

sub

isLoaded

open override val isLoaded: Boolean