Class PrecompiledComputeOps

Inheritance

Object

ReferenceCPUOps

ReferenceComputeOps

ComputeOps

PrecompiledComputeOps

Inherited Members

ComputeOps.BestKernel(ComputeKernelLibrary.Entry[])

ComputeOps.GlobalMaxPool2D(Tensor)

ComputeOps.GlobalAvgPool2D(Tensor)

ComputeOps.GlobalAvgVariancePool2D(Tensor)

ComputeOps.GlobalPool2D(String, String, Tensor)

ComputeOps.Normalization(Tensor, Tensor, Tensor, Int32, Int32, Single)

ComputeOps.RandomNormal(TensorShape, Single, Single, Int32)

ComputeOps.RandomUniform(TensorShape, Single, Single, Int32)

ComputeOps.Concat(Tensor[], Int32)

ComputeOps.ApplyPadding(Tensor, Int32[], String, Single)

ComputeOps.LogicalNot(Tensor)

ReferenceComputeOps.Pin(Tensor)

ReferenceComputeOps.WaitForCompletion(Tensor)

ReferenceComputeOps.SetTensor(ComputeFunc, String, Tensor)

ReferenceComputeOps.NewTensor(ComputeFunc, String, TensorShape)

ReferenceComputeOps.Dispatch(ComputeFunc, TensorShape, Int32, Int32, Int32, String)

ReferenceComputeOps.TextureToTensorData(TextureAsTensorData, String)

ReferenceComputeOps.TensorToRenderTexture(Tensor, RenderTexture, Int32, Int32, Single, Single)

ReferenceComputeOps.IDivC(Int32, Int32)

ReferenceComputeOps.Border2D(Tensor, Int32[], Single)

ReferenceComputeOps.Pad2DEdge(Tensor, Int32[])

ReferenceComputeOps.Pad2DReflect(Tensor, Int32[])

ReferenceComputeOps.Pad2DSymmetric(Tensor, Int32[])

ReferenceComputeOps.MaxPool2D(Tensor, Int32[], Int32[], Int32[])

ReferenceComputeOps.AvgPool2D(Tensor, Int32[], Int32[], Int32[])

ReferenceComputeOps.GlobalPool2D(String, Tensor)

ReferenceComputeOps.Dropout(Tensor, Single)

ReferenceComputeOps.Relu(Tensor)

ReferenceComputeOps.Selu(Tensor, Single, Single)

ReferenceComputeOps.Neg(Tensor)

ReferenceComputeOps.Swish(Tensor)

ReferenceComputeOps.Tanh(Tensor)

ReferenceComputeOps.Sigmoid(Tensor)

ReferenceComputeOps.Elu(Tensor, Single)

ReferenceComputeOps.Relu6(Tensor)

ReferenceComputeOps.LeakyRelu(Tensor, Single)

ReferenceComputeOps.Exp(Tensor)

ReferenceComputeOps.Log(Tensor)

ReferenceComputeOps.Sqrt(Tensor)

ReferenceComputeOps.Pow(Tensor, Single)

ReferenceComputeOps.Clip(Tensor, Single, Single)

ReferenceComputeOps.Gather(Tensor[], Int32)

ReferenceComputeOps.Add(Tensor[])

ReferenceComputeOps.Sub(Tensor[])

ReferenceComputeOps.Mul(Tensor[])

ReferenceComputeOps.Div(Tensor[])

ReferenceComputeOps.Pow(Tensor[])

ReferenceComputeOps.Min(Tensor[])

ReferenceComputeOps.Max(Tensor[])

ReferenceComputeOps.Greater(Tensor, Tensor)

ReferenceComputeOps.GreaterEqual(Tensor, Tensor)

ReferenceComputeOps.Less(Tensor, Tensor)

ReferenceComputeOps.LessEqual(Tensor, Tensor)

ReferenceComputeOps.Equal(Tensor, Tensor)

ReferenceComputeOps.LogicalOr(Tensor, Tensor)

ReferenceComputeOps.LogicalAnd(Tensor, Tensor)

ReferenceComputeOps.LogicalXor(Tensor, Tensor)

ReferenceComputeOps.Reduce(String, Tensor, Int32)

ReferenceComputeOps.ReduceMin(Tensor, Int32)

ReferenceComputeOps.ReduceMax(Tensor, Int32)

ReferenceComputeOps.ReduceSum(Tensor, Int32)

ReferenceComputeOps.ReduceMean(Tensor, Int32)

ReferenceComputeOps.ReduceProd(Tensor, Int32)

ReferenceComputeOps.Prepare(Tensor)

ReferenceCPUOps.NewTensor(TensorShape, String)

ReferenceCPUOps.NewTensorLike(Tensor)

ReferenceCPUOps.NewTensor(Int32, Int32, String)

ReferenceCPUOps.NewTensor(Int32, Int32, Int32, Int32, String)

ReferenceCPUOps.ResetAllocator(Boolean)

ReferenceCPUOps.MatMul(Tensor, Boolean, Tensor, Boolean)

ReferenceCPUOps.LRN(Tensor, Single, Single, Single, Int32)

ReferenceCPUOps.Bernoulli(Single)

ReferenceCPUOps.Gaussian(Single, Single)

ReferenceCPUOps.Multinomial(Tensor, Int32, Int32)

ReferenceCPUOps.OneHot(Tensor, Int32, Single, Single)

ReferenceCPUOps.Abs(Tensor)

ReferenceCPUOps.Ceil(Tensor)

ReferenceCPUOps.Floor(Tensor)

ReferenceCPUOps.Reciprocal(Tensor)

ReferenceCPUOps.StridedSlice(Tensor, Int32[], Int32[], Int32[])

ReferenceCPUOps.Tile(Tensor, Int32[])

ReferenceCPUOps.Mean(Tensor[])

ReferenceCPUOps.Flatten(Tensor)

ReferenceCPUOps.Reshape(Tensor, TensorShape)

ReferenceCPUOps.Transpose(Tensor)

Namespace: Barracuda

Syntax

public class PrecompiledComputeOps : ComputeOps, IOps, IModelCompiler

Constructors

PrecompiledComputeOps(ComputeShader[], ComputeShader, ITensorAllocator, Boolean)

Declaration

public PrecompiledComputeOps(ComputeShader[] kernels, ComputeShader referenceKernel, ITensorAllocator allocator = null, bool verbose = false)

Parameters

Type	Name	Description
ComputeShader[]	kernels
ComputeShader	referenceKernel
ITensorAllocator	allocator
Boolean	verbose

Fields

_Alpha

Declaration

public static int _Alpha

Field Value

Type	Description
Int32

_Beta

Declaration

public static int _Beta

Field Value

Type	Description
Int32

_DataB

Declaration

public static int _DataB

Field Value

Type	Description
Int32

_DataK

Declaration

public static int _DataK

Field Value

Type	Description
Int32

_DataO

Declaration

public static int _DataO

Field Value

Type	Description
Int32

_DataW

Declaration

public static int _DataW

Field Value

Type	Description
Int32

_DataWBK

Declaration

public static int _DataWBK

Field Value

Type	Description
Int32

_DataX

Declaration

public static int _DataX

Field Value

Type	Description
Int32

_DeclB

Declaration

public static ComputeFunc.TensorDecl _DeclB

Field Value

Type	Description
ComputeFunc.TensorDecl

_DeclK

Declaration

public static ComputeFunc.TensorDecl _DeclK

Field Value

Type	Description
ComputeFunc.TensorDecl

_DeclO

Declaration

public static ComputeFunc.TensorDecl _DeclO

Field Value

Type	Description
ComputeFunc.TensorDecl

_DeclW

Declaration

public static ComputeFunc.TensorDecl _DeclW

Field Value

Type	Description
ComputeFunc.TensorDecl

_DeclX

Declaration

public static ComputeFunc.TensorDecl _DeclX

Field Value

Type	Description
ComputeFunc.TensorDecl

_Pad

Declaration

public static int _Pad

Field Value

Type	Description
Int32

_Pool

Declaration

public static int _Pool

Field Value

Type	Description
Int32

_Stride

Declaration

public static int _Stride

Field Value

Type	Description
Int32

Methods

Activation(String, Tensor, Single, Single)

Declaration

protected override Tensor Activation(string kernelName, Tensor X, float alpha = 0F, float beta = 0F)

Parameters

Type	Name	Description
String	kernelName
Tensor	X
Single	alpha
Single	beta

Returns

Type	Description
Tensor

Overrides

ComputeOps.Activation(String, Tensor, Single, Single)

CalcModelWithInputsHashCode(Model, IDictionary<String, TensorShape>)

Declaration

protected int CalcModelWithInputsHashCode(Model model, IDictionary<string, TensorShape> inputShapes)

Parameters

Type	Name	Description
Model	model
IDictionary<String, TensorShape>	inputShapes

Returns

Type	Description
Int32

Conv2D(Tensor, Tensor, Tensor, Int32[], Int32[])

Declaration

public override Tensor Conv2D(Tensor X, Tensor K, Tensor B, int[] stride, int[] pad)

Parameters

Type	Name	Description
Tensor	X
Tensor	K
Tensor	B
Int32[]	stride
Int32[]	pad

Returns

Type	Description
Tensor

Overrides

ComputeOps.Conv2D(Tensor, Tensor, Tensor, Int32[], Int32[])

Conv2DTrans(Tensor, Tensor, Tensor, Int32[], Int32[], Int32[])

Declaration

public override Tensor Conv2DTrans(Tensor X, Tensor K, Tensor B, int[] stride, int[] pad, int[] outputAdjustment)

Parameters

Type	Name	Description
Tensor	X
Tensor	K
Tensor	B
Int32[]	stride
Int32[]	pad
Int32[]	outputAdjustment

Returns

Type	Description
Tensor

Overrides

ComputeOps.Conv2DTrans(Tensor, Tensor, Tensor, Int32[], Int32[], Int32[])

Dense(Tensor, Tensor, Tensor)

Declaration

public override Tensor Dense(Tensor X, Tensor W, Tensor B)

Parameters

Type	Name	Description
Tensor	X
Tensor	W
Tensor	B

Returns

Type	Description
Tensor

Overrides

ComputeOps.Dense(Tensor, Tensor, Tensor)

DepthwiseConv2D(Tensor, Tensor, Tensor, Int32[], Int32[])

Declaration

public override Tensor DepthwiseConv2D(Tensor X, Tensor K, Tensor B, int[] stride, int[] pad)

Parameters

Type	Name	Description
Tensor	X
Tensor	K
Tensor	B
Int32[]	stride
Int32[]	pad

Returns

Type	Description
Tensor

Overrides

ComputeOps.DepthwiseConv2D(Tensor, Tensor, Tensor, Int32[], Int32[])

ElementwiseWithBroadcast(String, Tensor[])

Declaration

public override Tensor ElementwiseWithBroadcast(string kernelName, Tensor[] tensors)

Parameters

Type	Name	Description
String	kernelName
Tensor[]	tensors

Returns

Type	Description
Tensor

Overrides

ComputeOps.ElementwiseWithBroadcast(String, Tensor[])

LogSoftmax(Tensor)

Declaration

public override Tensor LogSoftmax(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ComputeOps.LogSoftmax(Tensor)

Pool2D(String, Tensor, Int32[], Int32[], Int32[])

Declaration

protected override Tensor Pool2D(string kernelName, Tensor X, int[] pool, int[] stride, int[] pad)

Parameters

Type	Name	Description
String	kernelName
Tensor	X
Int32[]	pool
Int32[]	stride
Int32[]	pad

Returns

Type	Description
Tensor

Overrides

ComputeOps.Pool2D(String, Tensor, Int32[], Int32[], Int32[])

PreExecuteLayer(Layer, Tensor[])

Declaration

public virtual void PreExecuteLayer(Layer layer, Tensor[] inputs)

Parameters

Type	Name	Description
Layer	layer
Tensor[]	inputs

Implements

IModelCompiler.PreExecuteLayer(Layer, Tensor[])

PRelu(Tensor, Tensor)

Declaration

public override Tensor PRelu(Tensor X, Tensor S)

Parameters

Type	Name	Description
Tensor	X
Tensor	S

Returns

Type	Description
Tensor

Overrides

ComputeOps.PRelu(Tensor, Tensor)

PrepareModel(Model, IDictionary<String, TensorShape>)

Declaration

public virtual void PrepareModel(Model model, IDictionary<string, TensorShape> inputShapes)

Parameters

Type	Name	Description
Model	model
IDictionary<String, TensorShape>	inputShapes

Implements

IModelCompiler.PrepareModel(Model, IDictionary<String, TensorShape>)

ScaleBias(Tensor, Tensor, Tensor)

Declaration

public override Tensor ScaleBias(Tensor X, Tensor S, Tensor B)

Parameters

Type	Name	Description
Tensor	X
Tensor	S
Tensor	B

Returns

Type	Description
Tensor

Overrides

ComputeOps.ScaleBias(Tensor, Tensor, Tensor)

Softmax(Tensor)

Declaration

public override Tensor Softmax(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ComputeOps.Softmax(Tensor)

Upsample2D(Tensor, Int32[])

Declaration

public override Tensor Upsample2D(Tensor X, int[] size)

Parameters

Type	Name	Description
Tensor	X
Int32[]	size

Returns

Type	Description
Tensor

Overrides

ComputeOps.Upsample2D(Tensor, Int32[])