com.komputation.cuda.optimization.CudaUpdateRule.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of komputation Show documentation

Komputation is a neural network framework for the JVM written in the Kotlin programming language.

There is a newer version: 0.12.5

package com.komputation.cuda.optimization

import jcuda.Pointer

interface CudaUpdateRule {

    fun denseUpdate(
        count : Int,
        pointerToParameters: Pointer,
        pointerToGradient: Pointer)

    fun sparseUpdate(
        maximumParameters : Int,
        pointerToIndices: Pointer,
        pointerToCounts: Pointer,
        pointerToParameters: Pointer,
        pointerToGradient: Pointer)


}