kernels

package
v0.0.0-...-c9f06ed Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: May 13, 2020 License: MIT Imports: 8 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func DivUp 

func DivUp(a, b int32) int32

DivUp is same function from tensorflow

func DivUpUint32 

func DivUpUint32(a, b uint32) uint32

DivUpUint32 does it in uint 32

func LoadPTXFile 

func LoadPTXFile(directory, filename string) string

LoadPTXFile Loads the ptx file

func MakeMakeFile 

func MakeMakeFile(directory string, dotCUname string, device Device) string

MakeMakeFile Makes the make file

func SimpleGridCalculator 

func SimpleGridCalculator(blocksize uint32, size uint32) uint32

SimpleGridCalculator will output the size of the grid given the size of the block. size is the number of values. blocksizelllll should be multiples of 32!

Types

type Device 

type Device interface {
	Major() (int, error)
	Minor() (int, error)
}

Device just has to return those two things in order to compute the kernels

type XtraKerns 

type XtraKerns struct {
}

XtraKerns returns the kernel names through methods

func (XtraKerns) AdaDelta 

func (t XtraKerns) AdaDelta() string

AdaDelta ..

func (XtraKerns) AdaDeltaFP16 

func (t XtraKerns) AdaDeltaFP16() string

AdaDeltaFP16 ..

func (XtraKerns) AdaGrad 

func (t XtraKerns) AdaGrad() string

AdaGrad ..

func (XtraKerns) AdaGradFP16 

func (t XtraKerns) AdaGradFP16() string

AdaGradFP16 ..

func (XtraKerns) Adam 

func (t XtraKerns) Adam() string

Adam ..

func (XtraKerns) AdamFP16 

func (t XtraKerns) AdamFP16() string

AdamFP16 ..

func (XtraKerns) ConcatNCHWEX 

func (t XtraKerns) ConcatNCHWEX() string

ConcatNCHWEX - Does a concat forward with multiple inputs

func (XtraKerns) ConcatNCHWEXHalf 

func (t XtraKerns) ConcatNCHWEXHalf() string

ConcatNCHWEXHalf - Does a concat forward with multiple inputs

func (XtraKerns) ConcatNHWCEX 

func (t XtraKerns) ConcatNHWCEX() string

ConcatNHWCEX - Does a concat backward with multiple outputs

func (XtraKerns) ConcatNHWCEXHalf 

func (t XtraKerns) ConcatNHWCEXHalf() string

ConcatNHWCEXHalf - Does a concat backward with multiple outputs

func (XtraKerns) L1L2 

func (t XtraKerns) L1L2() string

L1L2 ..

func (XtraKerns) L1L2FP16 

func (t XtraKerns) L1L2FP16() string

L1L2FP16 ..

func (XtraKerns) LeakyBackward 

func (t XtraKerns) LeakyBackward() string

LeakyBackward activation function Relu but negatives get a reduced value

func (XtraKerns) LeakyBackwardAlpha 

func (t XtraKerns) LeakyBackwardAlpha() string

LeakyBackwardAlpha activation function Relu but negatives get a reduced value and function gets the ----- result = alpha * activationfunc()

func (XtraKerns) LeakyBackwardAlphaBeta 

func (t XtraKerns) LeakyBackwardAlphaBeta() string

LeakyBackwardAlphaBeta activation function Relu but negatives get a reduced value and function gets the ----- result = alpha * currentresult + beta * previousresult

func (XtraKerns) LeakyBackwardAlphaBetaFP16 

func (t XtraKerns) LeakyBackwardAlphaBetaFP16() string

LeakyBackwardAlphaBetaFP16 activation function Relu but negatives get a reduced value and function gets the ----- result = alpha * currentresult + beta * previousresult

func (XtraKerns) LeakyBackwardAlphaFP16 

func (t XtraKerns) LeakyBackwardAlphaFP16() string

LeakyBackwardAlphaFP16 activation function Relu but negatives get a reduced value and function gets the ----- result = alpha * activationfunc()

func (XtraKerns) LeakyBackwardFP16 

func (t XtraKerns) LeakyBackwardFP16() string

LeakyBackwardFP16 activation function Relu but negatives get a reduced value

func (XtraKerns) LeakyForward 

func (t XtraKerns) LeakyForward() string

LeakyForward activation function Relu but negatives get a reduced value

func (XtraKerns) LeakyForwardAlpha 

func (t XtraKerns) LeakyForwardAlpha() string

LeakyForwardAlpha activation function Relu but negatives get a reduced value result = alpha * activationfunc()

func (XtraKerns) LeakyForwardAlphaBeta 

func (t XtraKerns) LeakyForwardAlphaBeta() string

LeakyForwardAlphaBeta activation function Relu but negatives get a reduced value and function gets the ----- result = alpha * currentresult + beta * previousresult

func (XtraKerns) LeakyForwardAlphaBetaFP16 

func (t XtraKerns) LeakyForwardAlphaBetaFP16() string

LeakyForwardAlphaBetaFP16 activation function Relu but negatives get a reduced value and function gets the ----- result = alpha * currentresult + beta * previousresult

func (XtraKerns) LeakyForwardAlphaFP16 

func (t XtraKerns) LeakyForwardAlphaFP16() string

LeakyForwardAlphaFP16 activation function Relu but negatives get a reduced value result = alpha * activationfunc()

func (XtraKerns) LeakyForwardFP16 

func (t XtraKerns) LeakyForwardFP16() string

LeakyForwardFP16 activation function Relu but negatives get a reduced value

func (XtraKerns) MSELoss 

func (t XtraKerns) MSELoss() string

MSELoss Mean Squared Error Loss

func (XtraKerns) MSELossFP16 

func (t XtraKerns) MSELossFP16() string

MSELossFP16 Mean Squared Error Loss

func (XtraKerns) NearestNeighborNCHW 

func (t XtraKerns) NearestNeighborNCHW() string

NearestNeighborNCHW Resize NearestNeightbor for NCHW tensors

func (XtraKerns) NearestNeighborNCHWBack 

func (t XtraKerns) NearestNeighborNCHWBack() string

NearestNeighborNCHWBack Resize NearestNeightbor for NCHW tensors and accumulates gradients

func (XtraKerns) NearestNeighborNCHWBackFP16 

func (t XtraKerns) NearestNeighborNCHWBackFP16() string

NearestNeighborNCHWBackFP16 Resize NearestNeightbor for NCHW tensors and accumulates gradients

func (XtraKerns) NearestNeighborNCHWFP16 

func (t XtraKerns) NearestNeighborNCHWFP16() string

NearestNeighborNCHWFP16 Resize NearestNeightbor for NCHW tensors

func (XtraKerns) NearestNeighborNHWC 

func (t XtraKerns) NearestNeighborNHWC() string

NearestNeighborNHWC Resize NearestNeightbor for NHWC tensors

func (XtraKerns) NearestNeighborNHWCBack 

func (t XtraKerns) NearestNeighborNHWCBack() string

NearestNeighborNHWCBack Resize NearestNeightbor for NHWC tensors and accumulates gradients

func (XtraKerns) NearestNeighborNHWCBackFP16 

func (t XtraKerns) NearestNeighborNHWCBackFP16() string

NearestNeighborNHWCBackFP16 Resize NearestNeightbor for NHWC tensors and accumulates gradients

func (XtraKerns) NearestNeighborNHWCFP16 

func (t XtraKerns) NearestNeighborNHWCFP16() string

NearestNeighborNHWCFP16 Resize NearestNeightbor for NHWC tensors

func (XtraKerns) PreluBackward 

func (t XtraKerns) PreluBackward() string

PreluBackward Not tested

func (XtraKerns) PreluBackwardFP16 

func (t XtraKerns) PreluBackwardFP16() string

PreluBackwardFP16 Not tested

func (XtraKerns) PreluForward 

func (t XtraKerns) PreluForward() string

PreluForward Not tested

func (XtraKerns) PreluForwardFP16 

func (t XtraKerns) PreluForwardFP16() string

PreluForwardFP16 Not tested

func (XtraKerns) ShapeToBatch4DNCHW 

func (t XtraKerns) ShapeToBatch4DNCHW() string

ShapeToBatch4DNCHW transfers HW to batch and vice versa for NCHW tensors

func (XtraKerns) ShapeToBatch4DNHWC 

func (t XtraKerns) ShapeToBatch4DNHWC() string

ShapeToBatch4DNHWC transfer HW to batch and vice versa through windows for NHWC tensors

func (XtraKerns) ShapetoBatch4DNCHWFP16 

func (t XtraKerns) ShapetoBatch4DNCHWFP16() string

ShapetoBatch4DNCHWFP16 transfers HW to batch and vice versa for NCHW tensors

func (XtraKerns) ShapetoBatch4DNHWCFP16 

func (t XtraKerns) ShapetoBatch4DNHWCFP16() string

ShapetoBatch4DNHWCFP16 transfer HW to batch and vice versa through windows for NHWC tensors

func (XtraKerns) SoftMaxAverageLoss 

func (t XtraKerns) SoftMaxAverageLoss() string

SoftMaxAverageLoss

func (XtraKerns) SwapEveryOther 

func (t XtraKerns) SwapEveryOther() string

SwapEveryOther allows the user to swap batches between to tensors Either the even or the odd tensors.

func (XtraKerns) SwapEveryOtherFP16 

func (t XtraKerns) SwapEveryOtherFP16() string

SwapEveryOtherFP16 does SwapEveryOther in fp16

func (XtraKerns) SwapUpperLower 

func (t XtraKerns) SwapUpperLower() string

SwapUpperLower takes to tensors and swaps the upper or lower batches between the two tensors

func (XtraKerns) SwapUpperLowerFP16 

func (t XtraKerns) SwapUpperLowerFP16() string

SwapUpperLowerFP16 takes to tensors and swaps the upper or lower batches between the two tensors

func (XtraKerns) ThreshBackward 

func (t XtraKerns) ThreshBackward() string

ThreshBackward Not tested

func (XtraKerns) ThreshBackwardFP16 

func (t XtraKerns) ThreshBackwardFP16() string

ThreshBackwardFP16 Not tested

func (XtraKerns) ThreshForward 

func (t XtraKerns) ThreshForward() string

ThreshForward Not tested

func (XtraKerns) ThreshForwardFP16 

func (t XtraKerns) ThreshForwardFP16() string

ThreshForwardFP16 Not tested

func (XtraKerns) Transpose 

func (t XtraKerns) Transpose() string

Transpose switches values around from one dimention to the other

func (XtraKerns) TransposeFP16 

func (t XtraKerns) TransposeFP16() string

TransposeFP16 switches values around from one dimention to the other

Source Files

View all Source files

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.