Class PrecompiledComputeOps

Inheritance

Object

ReferenceCPUOps

ReferenceComputeOps

ComputeOps

PrecompiledComputeOps

Inherited Members

ComputeOps.GlobalAvgVariancePool2D(Tensor)

ComputeOps.GlobalPool2D(String, String, Tensor)

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

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

ComputeOps.GetInputTensorStridesOnDevice(TensorShape)

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

ComputeOps.LogicalNot(Tensor)

ComputeOps.CopyAndReshape_NCHW(Tensor, TensorShape)

ComputeOps.CopyAndReshape(Tensor, TensorShape)

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.MatMul(Tensor, Boolean, Tensor, Boolean)

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.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.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.Bernoulli(Single)

ReferenceCPUOps.Gaussian(Single, Single)

ReferenceCPUOps.Multinomial(Tensor, Int32, Int32)

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

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 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)

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

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

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

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

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

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

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

ComputeOps.ElementwiseWithBroadcast(String, Tensor[])

GlobalAvgPool2D(Tensor)

Declaration

public override Tensor GlobalAvgPool2D(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ComputeOps.GlobalAvgPool2D(Tensor)

GlobalMaxPool2D(Tensor)

Declaration

public override Tensor GlobalMaxPool2D(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ComputeOps.GlobalMaxPool2D(Tensor)

LogSoftmax(Tensor)

Declaration

public override Tensor LogSoftmax(Tensor X)

Parameters

Type	Name	Description
Tensor	X

Returns

Type	Description
Tensor

Overrides

ComputeOps.LogSoftmax(Tensor)

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

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

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

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

ResetAllocator(Boolean)

Declaration

public override void ResetAllocator(bool keepCachedMemory = true)

Parameters

Type	Name	Description
Boolean	keepCachedMemory

Overrides

ReferenceCPUOps.ResetAllocator(Boolean)

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

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