Class ComputeKernelLibrary

Inheritance

Object

ComputeKernelLibrary

Namespace: Barracuda

Syntax

public sealed class ComputeKernelLibrary

Methods

Activation(TensorShape, TensorShape, String)

Declaration

public static ComputeKernelLibrary.Entry[] Activation(TensorShape X, TensorShape O, string kernelName)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O
String	kernelName

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Broadcast(TensorShape, TensorShape, String)

Declaration

public static ComputeKernelLibrary.Entry[] Broadcast(TensorShape X, TensorShape O, string kernelName)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O
String	kernelName

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Conv2D(TensorShape, TensorShape, TensorShape, Int32[], Int32[])

Declaration

public static ComputeKernelLibrary.Entry[] Conv2D(TensorShape X, TensorShape K, TensorShape O, int[] stride, int[] pad)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	K
TensorShape	O
Int32[]	stride
Int32[]	pad

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Conv2DTrans(TensorShape, TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] Conv2DTrans(TensorShape X, TensorShape K, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	K
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Copy(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] Copy(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Dense(TensorShape, TensorShape, TensorShape, Int32)

Declaration

public static ComputeKernelLibrary.Entry[] Dense(TensorShape X, TensorShape W, TensorShape O, int type)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	W
TensorShape	O
Int32	type

Returns

Type	Description
ComputeKernelLibrary.Entry[]

DepthwiseConv2D(TensorShape, TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] DepthwiseConv2D(TensorShape X, TensorShape K, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	K
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

GlobalPool2D(TensorShape, TensorShape, String)

Declaration

public static ComputeKernelLibrary.Entry[] GlobalPool2D(TensorShape X, TensorShape O, string kernelName)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O
String	kernelName

Returns

Type	Description
ComputeKernelLibrary.Entry[]

IDivC(Int32, Int32)

Declaration

public static int IDivC(int v, int div)

Parameters

Type	Name	Description
Int32	v
Int32	div

Returns

Type	Description
Int32

LogSoftmax(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] LogSoftmax(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Normalization(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] Normalization(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

NormalizationTail(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] NormalizationTail(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Padding(TensorShape, TensorShape, String)

Declaration

public static ComputeKernelLibrary.Entry[] Padding(TensorShape X, TensorShape O, string kernelName)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O
String	kernelName

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Pool2D(TensorShape, TensorShape, String)

Declaration

public static ComputeKernelLibrary.Entry[] Pool2D(TensorShape X, TensorShape O, string kernelName)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O
String	kernelName

Returns

Type	Description
ComputeKernelLibrary.Entry[]

PRelu(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] PRelu(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

ScaleBias(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] ScaleBias(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Softmax(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] Softmax(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]

Upsample2D(TensorShape, TensorShape)

Declaration

public static ComputeKernelLibrary.Entry[] Upsample2D(TensorShape X, TensorShape O)

Parameters

Type	Name	Description
TensorShape	X
TensorShape	O

Returns

Type	Description
ComputeKernelLibrary.Entry[]