Gpu

object Gpu : ExternalLib

Functions

add

external fun add(lhs: Long, rhs: Long): LongArray

addGradLhs

external fun addGradLhs(seed: Long, lhs: Long, forwardRes: Long): Long

addGradRhs

external fun addGradRhs(seed: Long, rhs: Long, forwardRes: Long): Long

avgPool

external fun avgPool(x: Long, poolHeight: Int, poolWidth: Int): LongArray

avgPoolGrad

external fun avgPoolGrad(seed: Long, x: Long, forwardRes: Long): Long

batchNorm2d

external fun batchNorm2d(input: Long, scaleShift: Long, runningMean: Long, runningVariance: Long, momentum: Float): LongArray

batchNorm2dGradInput

external fun batchNorm2dGradInput(seed: Long, input: Long, forwardRes: Long): Long

batchNorm2dGradScaleShift

external fun batchNorm2dGradScaleShift(seed: Long, scaleShift: Long, forwardRes: Long): Long

broadcastTo

external fun broadcastTo(seed: Long, newShape: IntArray): LongArray

broadcastToGrad

external fun broadcastToGrad(seed: Long, forwardArg: Long, forwardRes: Long): Long

conv2d

external fun conv2d(images: Long, filters: Long, strides: IntArray, padding: IntArray): LongArray

conv2dGradFilters

external fun conv2dGradFilters(seed: Long, filters: Long, forwardRes: Long): Long

conv2dGradImages

external fun conv2dGradImages(seed: Long, images: Long, forwardRes: Long): Long

deleteHandle

external fun deleteHandle(handle: Long)

div

external fun div(lhs: Long, rhs: Long): Long

getAllocatedBytes

external fun getAllocatedBytes(): Long

getFloatData

external fun getFloatData(handle: Long): FloatArray

getShape

external fun getShape(handle: Long): IntArray

isub

external fun isub(lhs: Long, rhs: Long): Long

logSoftmax

external fun logSoftmax(x: Long, axis: Int): LongArray

logSoftmaxGrad

external fun logSoftmaxGrad(seed: Long, x: Long, forwardRes: Long): Long

matmul

external fun matmul(lhs: Long, rhs: Long): LongArray

matmulGradLhs

external fun matmulGradLhs(seed: Long, lhs: Long, forwardRes: Long): Long

matmulGradRhs

external fun matmulGradRhs(seed: Long, rhs: Long, forwardRes: Long): Long

maxPool

external fun maxPool(x: Long, poolHeight: Int, poolWidth: Int): LongArray

maxPoolGrad

external fun maxPoolGrad(seed: Long, x: Long, forwardRes: Long): Long

nllLoss

external fun nllLoss(x: Long, labels: Long): LongArray

nllLossGradLabels

external fun nllLossGradLabels(seed: Long, labels: Long, forwardRes: Long): Long

nllLossGradX

external fun nllLossGradX(seed: Long, x: Long, forwardRes: Long): Long

putFloatTensor

external fun putFloatTensor(shape: IntArray, data: FloatArray): Long

relu

external fun relu(handle: Long): LongArray

reluGrad

external fun reluGrad(seed: Long, forwardArg: Long, forwardRes: Long): Long

reshape

external fun reshape(handle: Long, newShape: IntArray): LongArray

reshapeGrad

external fun reshapeGrad(seed: Long, forwardArg: Long, forwardRes: Long): Long

sub

external fun sub(lhs: Long, rhs: Long): Long

sum

external fun sum(handle: Long, axes: IntArray, keepDims: Boolean): LongArray

sumGrad

external fun sumGrad(seed: Long, forwardArg: Long, forwardRes: Long): Long

times

external fun times(lhs: Long, rhs: Long): Long

zeros

external fun zeros(shape: IntArray): Long

Properties

isLoaded

open override val isLoaded: Boolean