Class ComputeOps

Inheritance

Object

ReferenceCPUOps

ReferenceComputeOps

ComputeOps

PrecompiledComputeOps

Inherited Members

ReferenceComputeOps.Pin(Tensor)

ReferenceComputeOps.SetTensor(ComputeFunc, String, Tensor)

ReferenceComputeOps.NewTensor(ComputeFunc, String, TensorShape)

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

ReferenceComputeOps.TensorToRenderTexture(Tensor, RenderTexture, Int32, Int32, Vector4, Vector4, Texture3D)

ReferenceComputeOps.ShouldFlattenInputForDenseLayer(TensorShape)

ReferenceComputeOps.IsFusedActivationSupported(Layer.FusedActivation)

ReferenceComputeOps.Resample2D(Tensor, Int32[], Boolean)

ReferenceComputeOps.DepthToSpace(Tensor, Int32[], Layer.DepthToSpaceMode)

ReferenceComputeOps.SpaceToDepth(Tensor, Int32[])

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

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

ReferenceComputeOps.GlobalPool2D(String, Tensor)

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

ReferenceComputeOps.Pad2DReflect(Tensor, Int32[])

ReferenceComputeOps.Pad2DSymmetric(Tensor, Int32[])

ReferenceComputeOps.Pad2DEdge(Tensor, Int32[])

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

ReferenceComputeOps.Dropout(Tensor, Single)

ReferenceComputeOps.Relu(Tensor)

ReferenceComputeOps.Tanh(Tensor)

ReferenceComputeOps.Sigmoid(Tensor)

ReferenceComputeOps.Relu6(Tensor)

ReferenceComputeOps.Elu(Tensor, Single)

ReferenceComputeOps.LeakyRelu(Tensor, Single)

ReferenceComputeOps.Selu(Tensor, Single, Single)

ReferenceComputeOps.Swish(Tensor)

ReferenceComputeOps.Abs(Tensor)

ReferenceComputeOps.Neg(Tensor)

ReferenceComputeOps.Ceil(Tensor)

ReferenceComputeOps.Clip(Tensor, Single, Single)

ReferenceComputeOps.Floor(Tensor)

ReferenceComputeOps.Reciprocal(Tensor)

ReferenceComputeOps.Pow(Tensor, Single)

ReferenceComputeOps.Exp(Tensor)

ReferenceComputeOps.Log(Tensor)

ReferenceComputeOps.Sqrt(Tensor)

ReferenceComputeOps.Acos(Tensor)

ReferenceComputeOps.Acosh(Tensor)

ReferenceComputeOps.Asin(Tensor)

ReferenceComputeOps.Asinh(Tensor)

ReferenceComputeOps.Atan(Tensor)

ReferenceComputeOps.Atanh(Tensor)

ReferenceComputeOps.Cos(Tensor)

ReferenceComputeOps.Cosh(Tensor)

ReferenceComputeOps.Sin(Tensor)

ReferenceComputeOps.Sinh(Tensor)

ReferenceComputeOps.Tan(Tensor)

ReferenceComputeOps.Expand(Tensor, TensorShape)

ReferenceComputeOps.Add(Tensor[])

ReferenceComputeOps.Sub(Tensor[])

ReferenceComputeOps.Mul(Tensor[])

ReferenceComputeOps.Div(Tensor[])

ReferenceComputeOps.Pow(Tensor[])

ReferenceComputeOps.Min(Tensor[])

ReferenceComputeOps.Max(Tensor[])

ReferenceComputeOps.Mean(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.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.Flatten(Tensor)

ReferenceComputeOps.Reshape(Tensor, TensorShape)

ReferenceComputeOps.Transpose(Tensor)

ReferenceComputeOps.Transpose(Tensor, Int32[])

ReferenceComputeOps.s_ConcatOffsets

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

ReferenceComputeOps.Tile(Tensor, Int32[])

ReferenceComputeOps.Gather(Tensor[], Int32)

ReferenceComputeOps.Copy(Tensor)

ReferenceComputeOps.Prepare(Tensor)

ReferenceCPUOps.NewTensor(TensorShape, String)

ReferenceCPUOps.NewTensorLike(Tensor)

ReferenceCPUOps.NewTensorLike(Tensor[])

ReferenceCPUOps.NewTensor(Int32, Int32, String)

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

ReferenceCPUOps.ResetAllocator(Boolean)

ReferenceCPUOps.Bernoulli(Single)

ReferenceCPUOps.Gaussian(Single, Single)

ReferenceCPUOps.Multinomial(Tensor, Int32, Int32)

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

ReferenceCPUOps.TopKIndices(Tensor, Int32, Int32, Boolean, Boolean)

ReferenceCPUOps.TopKValues(Tensor, Tensor, Int32)

Object.ToString()

Object.Equals(Object)

Object.Equals(Object, Object)

Object.ReferenceEquals(Object, Object)

Object.GetHashCode()

Object.GetType()

Object.MemberwiseClone()

Namespace: Unity.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)

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[], Layer.FusedActivation)

Declaration

public override Tensor Conv2D(Tensor X, Tensor K, Tensor B, int[] stride, int[] pad, Layer.FusedActivation fusedActivation)

Parameters

Type	Name	Description
Tensor	X
Tensor	K
Tensor	B
Int32[]	stride
Int32[]	pad
Layer.FusedActivation	fusedActivation

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.Conv2D(Tensor, Tensor, Tensor, Int32[], Int32[], Layer.FusedActivation)

Conv2DTrans(Tensor, Tensor, Tensor, Int32[], Int32[], Int32[], Layer.FusedActivation)

Declaration

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

Parameters

Type	Name	Description
Tensor	X
Tensor	K
Tensor	B
Int32[]	stride
Int32[]	pad
Int32[]	outputAdjustment
Layer.FusedActivation	fusedActivation

Returns

Type	Description
Tensor

Overrides

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

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)

CopyAndReshape_NCHW(Tensor, TensorShape)

Declaration

protected override Tensor CopyAndReshape_NCHW(Tensor X, TensorShape newShape)

Parameters

Type	Name	Description
Tensor	X
TensorShape	newShape

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.CopyAndReshape_NCHW(Tensor, TensorShape)

Dense(Tensor, Tensor, Tensor, Layer.FusedActivation)

Declaration

public override Tensor Dense(Tensor X, Tensor W, Tensor B, Layer.FusedActivation fusedActivation)

Parameters

Type	Name	Description
Tensor	X
Tensor	W
Tensor	B
Layer.FusedActivation	fusedActivation

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.Dense(Tensor, Tensor, Tensor, Layer.FusedActivation)

DepthwiseConv2D(Tensor, Tensor, Tensor, Int32[], Int32[], Layer.FusedActivation)

Declaration

public override Tensor DepthwiseConv2D(Tensor X, Tensor K, Tensor B, int[] stride, int[] pad, Layer.FusedActivation fusedActivation)

Parameters

Type	Name	Description
Tensor	X
Tensor	K
Tensor	B
Int32[]	stride
Int32[]	pad
Layer.FusedActivation	fusedActivation

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.DepthwiseConv2D(Tensor, Tensor, Tensor, Int32[], Int32[], Layer.FusedActivation)

ElementwiseWithBroadcast(String, Tensor[])

Declaration

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

Parameters

Type	Name	Description
String	kernelName
Tensor[]	tensors

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.ElementwiseWithBroadcast(String, Tensor[])

GetInputTensorStridesOnDevice(TensorShape, ComputeInfo.ChannelsOrder)

Declaration

protected int[] GetInputTensorStridesOnDevice(TensorShape shape, ComputeInfo.ChannelsOrder channelOrder)

Parameters

Type	Name	Description
TensorShape	shape
ComputeInfo.ChannelsOrder	channelOrder

Returns

Type	Description
Int32[]

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)

MatMul(Tensor, Boolean, Tensor, Boolean)

Declaration

public override Tensor MatMul(Tensor X, bool xTranspose, Tensor Y, bool yTranspose)

Parameters

Type	Name	Description
Tensor	X
Boolean	xTranspose
Tensor	Y
Boolean	yTranspose

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.MatMul(Tensor, Boolean, Tensor, Boolean)

Normalization(Tensor, Tensor, Tensor, Int32, Int32, Single, Layer.FusedActivation)

Declaration

public override Tensor Normalization(Tensor X, Tensor S, Tensor B, int pool, int axis, float epsilon, Layer.FusedActivation fusedActivation)

Parameters

Type	Name	Description
Tensor	X
Tensor	S
Tensor	B
Int32	pool
Int32	axis
Single	epsilon
Layer.FusedActivation	fusedActivation

Returns

Type	Description
Tensor

Overrides

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

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[], Boolean)

Declaration

public override Tensor Upsample2D(Tensor X, int[] scale, bool bilinear)

Parameters

Type	Name	Description
Tensor	X
Int32[]	scale
Boolean	bilinear

Returns

Type	Description
Tensor

Overrides

ReferenceComputeOps.Upsample2D(Tensor, Int32[], Boolean)