Class ComputeOps

Inheritance

Object

ReferenceCPUOps

ReferenceComputeOps

ComputeOps

PrecompiledComputeOps

Inherited Members

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.Copy(Tensor)

ReferenceCPUOps.Flatten(Tensor)

ReferenceCPUOps.Reshape(Tensor, TensorShape)

ReferenceCPUOps.Transpose(Tensor)

Namespace: Barracuda

Syntax

public class ComputeOps : ReferenceComputeOps, IOps

Constructors

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

Declaration

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

Parameters

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

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

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

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

Declaration

protected override Tensor ApplyPadding(Tensor X, int[] pad, string kernelName, float constant = 0F)

Parameters

Type	Name	Description
Tensor	X
Int32[]	pad
String	kernelName
Single	constant

Returns

Type	Description
Tensor

Overrides

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

BestKernel(ComputeKernelLibrary.Entry[])

Declaration

protected ComputeKernel BestKernel(ComputeKernelLibrary.Entry[] entrees)

Parameters

Type	Name	Description
ComputeKernelLibrary.Entry[]	entrees

Returns

Type	Description
ComputeKernel

Concat(Tensor[], Int32)

Declaration

public override Tensor Concat(Tensor[] tensors, int axis)

Parameters

Type	Name	Description
Tensor[]	tensors
Int32	axis

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.Concat(Tensor[], 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

ReferenceComputeOps.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

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

CopyAndReshape(Tensor, TensorShape)

Declaration

protected override Tensor CopyAndReshape(Tensor X, TensorShape newShape)

Parameters

Type	Name	Description
Tensor	X
TensorShape	newShape

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.CopyAndReshape(Tensor, TensorShape)

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

ReferenceComputeOps.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

ReferenceComputeOps.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

ReferenceComputeOps.ElementwiseWithBroadcast(String, Tensor[])

GlobalAvgPool2D(Tensor)

Declaration

public override Tensor GlobalAvgPool2D(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.GlobalAvgPool2D(Tensor)

GlobalAvgVariancePool2D(Tensor)

Declaration

public override Tensor GlobalAvgVariancePool2D(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.GlobalAvgVariancePool2D(Tensor)

GlobalMaxPool2D(Tensor)

Declaration

public override Tensor GlobalMaxPool2D(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.GlobalMaxPool2D(Tensor)

GlobalPool2D(String, String, Tensor)

Declaration

protected virtual Tensor GlobalPool2D(string smallKernelName, string globalKernelName, Tensor X)

Parameters

Type	Name	Description
String	smallKernelName
String	globalKernelName
Tensor	X

Returns

Type	Description
Tensor

LogicalNot(Tensor)

Declaration

public override Tensor LogicalNot(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.LogicalNot(Tensor)

LogSoftmax(Tensor)

Declaration

public override Tensor LogSoftmax(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.LogSoftmax(Tensor)

Normalization(Tensor, Tensor, Tensor, Int32, Int32, Single)

Declaration

public override Tensor Normalization(Tensor X, Tensor S, Tensor B, int pool, int axis, float epsilon)

Parameters

Type	Name	Description
Tensor	X
Tensor	S
Tensor	B
Int32	pool
Int32	axis
Single	epsilon

Returns

Type	Description
Tensor

Overrides

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

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

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

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

ReferenceComputeOps.PRelu(Tensor, Tensor)

RandomNormal(TensorShape, Single, Single, Int32)

Declaration

public override Tensor RandomNormal(TensorShape s, float mean, float scale, int seed)

Parameters

Type	Name	Description
TensorShape	s
Single	mean
Single	scale
Int32	seed

Returns

Type	Description
Tensor

Overrides

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

RandomUniform(TensorShape, Single, Single, Int32)

Declaration

public override Tensor RandomUniform(TensorShape s, float mean, float scale, int seed)

Parameters

Type	Name	Description
TensorShape	s
Single	mean
Single	scale
Int32	seed

Returns

Type	Description
Tensor

Overrides

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

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

ReferenceComputeOps.ScaleBias(Tensor, Tensor, Tensor)

Softmax(Tensor)

Declaration

public override Tensor Softmax(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.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

ReferenceComputeOps.Upsample2D(Tensor, Int32[])