Class GPUComputeBackend
Represents a GPUCompute backend ops.
Inherited Members
Namespace: Unity.Sentis
Assembly: solution.dll
Syntax
public class GPUComputeBackend : CPUBackend, IBackend, IDisposable
Constructors
Name | Description |
---|---|
GPUComputeBackend(ITensorAllocator) | Initializes and returns an instance of |
Properties
Name | Description |
---|---|
deviceType | Returns the |
Methods
Name | Description |
---|---|
Abs(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Abs(TensorInt, TensorInt) | Computes an output tensor by applying the element-wise |
Acos(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Acosh(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Add(TensorFloat, TensorFloat, TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Add(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
And(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
ArgMax(TensorFloat, TensorInt, int, bool, bool) | Computes the indices of the maximum elements of the input tensor along a given axis. |
ArgMax(TensorInt, TensorInt, int, bool, bool) | Computes the indices of the maximum elements of the input tensor along a given axis. |
ArgMin(TensorFloat, TensorInt, int, bool, bool) | Computes the indices of the minimum elements of the input tensor along a given axis. |
ArgMin(TensorInt, TensorInt, int, bool, bool) | Computes the indices of the minimum elements of the input tensor along a given axis. |
Asin(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Asinh(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Atan(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Atanh(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
AveragePool(TensorFloat, TensorFloat, int[], int[], int[]) | Calculates an output tensor by pooling the mean values of the input tensor across its spatial dimensions according to the given pool and stride values. |
BatchNormalization(TensorFloat, TensorFloat, TensorFloat, TensorFloat, TensorFloat, TensorFloat, float) | Computes the mean variance on the last dimension of the input tensor and normalizes it according to |
Bernoulli(TensorFloat, Tensor, float?) | Generates an output tensor with values 0 or 1 from a Bernoulli distribution. The input tensor contains the probabilities to use for generating the output values. |
Cast(Tensor, Tensor) | Computes the output tensor using an element-wise |
Ceil(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Celu(TensorFloat, TensorFloat, float) | Computes an output tensor by applying the element-wise |
Clip(TensorFloat, TensorFloat, float, float) | Computes an output tensor by applying the element-wise |
Clip(TensorInt, TensorInt, int, int) | Computes an output tensor by applying the element-wise |
CompressWithIndices(Tensor, TensorInt, Tensor, int, int) | Computes the output tensor by selecting slices from an input tensor according to the 'indices' tensor along an 'axis'. |
Concat(Tensor[], Tensor, int) | Calculates an output tensor by concatenating the input tensors along a given axis. |
Conv(TensorFloat, TensorFloat, TensorFloat, TensorFloat, int, Span<int>, Span<int>, Span<int>, FusableActivation) | Applies a convolution filter to an input tensor. |
ConvTranspose(TensorFloat, TensorFloat, TensorFloat, TensorFloat, Span<int>, Span<int>, Span<int>, FusableActivation) | Applies a transpose convolution filter to an input tensor. |
Cos(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Cosh(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
CumSum(TensorFloat, TensorFloat, int, bool, bool) | Performs the cumulative sum along a given axis. |
CumSum(TensorInt, TensorInt, int, bool, bool) | Performs the cumulative sum along a given axis. |
Dense(TensorFloat, TensorFloat, TensorFloat, TensorFloat, FusableActivation) | Performs a matrix multiplication operation: f(x, w, b) = X x W + B. This supports numpy-style broadcasting of input tensors. |
DepthToSpace(TensorFloat, TensorFloat, int, DepthToSpaceMode) | Computes the output tensor by permuting data from depth into blocks of spatial data. |
Div(TensorFloat, TensorFloat, TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Div(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Einsum(TensorFloat[], TensorFloat, TensorIndex[], TensorIndex, TensorIndex, TensorShape) | Performs an |
Elu(TensorFloat, TensorFloat, float) | Computes an output tensor by applying the element-wise |
Equal(TensorFloat, TensorFloat, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Equal(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Erf(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Exp(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Expand(Tensor, Tensor) | Calculates an output tensor by broadcasting the input tensor into a given shape. |
FMod(TensorFloat, TensorFloat, TensorFloat) | Performs an element-wise The sign of the remainder is the same as the sign of the dividend, as in C#. This supports numpy-style broadcasting of input tensors. |
FMod(TensorInt, TensorInt, TensorInt) | Performs an element-wise The sign of the remainder is the same as the sign of the dividend, as in C#. This supports numpy-style broadcasting of input tensors. |
Floor(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Gather(Tensor, TensorInt, Tensor, int) | Takes values from the input tensor indexed by the indices tensor along a given axis and concatenates them. |
GatherElements(Tensor, TensorInt, Tensor, int) | Takes values from the input tensor indexed by the indices tensor along a given axis and concatenates them. |
GatherND(Tensor, TensorInt, Tensor, int) | Takes slices of values from the batched input tensor indexed by the |
Gelu(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
GlobalAveragePool(TensorFloat, TensorFloat) | Calculates an output tensor by pooling the mean values of the input tensor across all of its spatial dimensions. The spatial dimensions of the output are size 1. |
GlobalAverageVariancePool(TensorFloat, TensorFloat, int) | Calculates an output tensor by pooling the mean and variance values of the input tensor across the spatial dimensions from a given axis. The spatial dimensions of the output are size 1. |
GlobalMaxPool(TensorFloat, TensorFloat) | Calculates an output tensor by pooling the maximum values of the input tensor across all of its spatial dimensions. The spatial dimensions of the output are size 1. |
Greater(TensorFloat, TensorFloat, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Greater(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
GreaterOrEqual(TensorFloat, TensorFloat, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
GreaterOrEqual(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
HardSigmoid(TensorFloat, TensorFloat, float, float) | Computes an output tensor by applying the element-wise |
HardSwish(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Hardmax(TensorFloat, TensorFloat, int) | Computes an output tensor by applying the |
InstanceNormalization(TensorFloat, TensorFloat, TensorFloat, TensorFloat, float) | Computes the mean variance on the spatial dimensions of the input tensor and normalizes them according to |
IsInf(TensorFloat, TensorInt, bool, bool) | Performs an element-wise |
IsNaN(TensorFloat, TensorInt) | Performs an element-wise |
LSTM(TensorFloat, TensorFloat, TensorFloat, TensorFloat, TensorInt, TensorFloat, TensorFloat, TensorFloat, TensorFloat, TensorFloat, TensorFloat, RnnDirection, RnnActivation[], float[], float[], bool, float, RnnLayout) | Generates an output tensor by computing a one-layer long short-term memory (LSTM) on an input tensor. |
LayerNormalization(TensorFloat, TensorFloat, TensorFloat, TensorFloat, float) | Computes the mean variance on the last dimension of the input tensor and normalizes it according to |
LeakyRelu(TensorFloat, TensorFloat, float) | Computes an output tensor by applying the element-wise |
Less(TensorFloat, TensorFloat, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Less(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
LessOrEqual(TensorFloat, TensorFloat, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
LessOrEqual(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Log(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
LogSoftmax(TensorFloat, TensorFloat, int) | Computes an output tensor by applying the |
MatMul(TensorFloat, TensorFloat, TensorFloat) | Performs a multi-dimensional matrix multiplication operation: f(a, b) = a x b. |
MatMul2D(TensorFloat, TensorFloat, TensorFloat, bool, bool) | Performs a matrix multiplication operation with optional transposes: f(a, b) = a' x b'. |
Max(TensorFloat[], TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Max(TensorInt[], TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
MaxPool(TensorFloat, TensorFloat, int[], int[], int[]) | Calculates an output tensor by pooling the maximum values of the input tensor across its spatial dimensions according to the given pool and stride values. |
Mean(TensorFloat[], TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
MemClear(Tensor) | Sets the entries of a tensor to 0. |
MemCopy(Tensor, Tensor) | Creates a copy of a given input tensor with the same shape and values. |
MemCopyStride(Tensor, Tensor, int, int, int, int, int, int) | Copy blocks of values from X to O, we copy 'count' blocks each of length 'length' values with initial offsets given by 'offsetX', 'offsetO' and with strides given by 'strideX', 'strideO' |
MemSet(TensorFloat, float) | Sets the entries of a tensor to a given fill value. |
MemSet(TensorInt, int) | Sets the entries of a tensor to a given fill value. |
Min(TensorFloat[], TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Min(TensorInt[], TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Mod(TensorInt, TensorInt, TensorInt) | Performs an element-wise The sign of the remainder is the same as the sign of the divisor, as in Python. This supports numpy-style broadcasting of input tensors. |
Mul(TensorFloat, TensorFloat, TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Mul(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Neg(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Neg(TensorInt, TensorInt) | Computes an output tensor by applying the element-wise |
NewTensor(TensorShape, DataType, AllocScope) | Allocates a new tensor with the internal allocator. |
Not(TensorInt, TensorInt) | Performs an element-wise |
OneHot(TensorInt, TensorFloat, int, int, float, float) | Generates a one-hot tensor with a given |
OneHot(TensorInt, TensorInt, int, int, int, int) | Generates a one-hot tensor with a given |
Or(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
PRelu(TensorFloat, TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Pad(TensorFloat, TensorFloat, ReadOnlySpan<int>, PadMode, float) | Calculates the output tensor by adding padding to the input tensor according to the given padding values and mode. |
PinToDevice(Tensor, bool) | Prepares |
Pow(TensorFloat, TensorFloat, TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Pow(TensorFloat, TensorInt, TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
RandomNormal(TensorFloat, float, float, float?) | Generates an output tensor of a given shape with random values in a normal distribution with given |
RandomUniform(TensorFloat, float, float, float?) | Generates an output tensor of a given shape with random values in a uniform distribution between a given |
Range(TensorFloat, float, float) | Generates a 1D output tensor where the values form an arithmetic progression defined by the |
Range(TensorInt, int, int) | Generates a 1D output tensor where the values form an arithmetic progression defined by the |
Reciprocal(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
ReduceL1(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceL1(TensorInt, TensorInt, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceL2(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceLogSum(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceLogSumExp(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceMax(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceMax(TensorInt, TensorInt, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceMean(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceMin(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceMin(TensorInt, TensorInt, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceProd(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceProd(TensorInt, TensorInt, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceSum(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceSum(TensorInt, TensorInt, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceSumExp(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | |
ReduceSumSquare(TensorFloat, TensorFloat, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
ReduceSumSquare(TensorInt, TensorInt, ReadOnlySpan<int>, bool) | Reduces an input tensor along the given axes using the |
Relu(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Relu6(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Reshape(Tensor, Tensor) | Calculates an output tensor by copying the data from the input tensor and using a given shape. The data from the input tensor is unchanged. |
Resize(TensorFloat, TensorFloat, ReadOnlySpan<float>, InterpolationMode, NearestMode, CoordTransformMode) | Calculates an output tensor by resampling the input tensor along the spatial dimensions with given scales. |
RoiAlign(TensorFloat, TensorFloat, TensorInt, TensorFloat, RoiPoolingMode, int, int, int, float) | Calculates an output tensor by pooling the input tensor across each region of interest given by the |
Round(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise If the fractional part is equal to 0.5, rounds to the nearest even integer. |
ScalarMad(TensorFloat, TensorFloat, float, float) | Performs an element-wise |
ScaleBias(TensorFloat, TensorFloat, TensorFloat, TensorFloat) | Computes the output tensor with an element-wise |
ScatterElements(Tensor, TensorInt, Tensor, Tensor, int, ScatterReductionMode) | Copies the input tensor and updates values at indexes specified by the
|
ScatterND(TensorFloat, TensorInt, TensorFloat, TensorFloat, ScatterReductionMode) | Copies the input tensor and updates values at indexes specified by the
|
ScatterND(TensorInt, TensorInt, TensorInt, TensorInt, ScatterReductionMode) | Copies the input tensor and updates values at indexes specified by the
|
Selu(TensorFloat, TensorFloat, float, float) | Computes an output tensor by applying the element-wise |
Shrink(TensorFloat, TensorFloat, float, float) | Computes an output tensor by applying the element-wise |
Sigmoid(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Sign(TensorFloat, TensorFloat) | Performs an element-wise |
Sign(TensorInt, TensorInt) | Performs an element-wise |
Sin(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
SinglePassLSTM(TensorFloat, TensorFloat, TensorFloat, TensorFloat, TensorInt, TensorFloat, TensorFloat, TensorFloat, TensorFloat, RnnActivation[], float[], float[], bool, float, bool, int, RnnLayout) | Computes a single pass LSTM either forward or reverse dirIndex and layout are used to calculate where to index the various tensors in bidirectional and batch first layout passes X has given layout W, R are cropped to single direction P, B are full number of directions Y has given layout and full number of directions (matches output of Layer) Y_h, Y_c are SequenceFirst layout and cropped to single direction HtxRT and XsixWT are temp vectors of the correct dimension for the intermediate results of the matmuls activations, activationAlpha and activationBeta have full number of dimensions |
Sinh(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Slice(Tensor, Tensor, ReadOnlySpan<int>, ReadOnlySpan<int>, ReadOnlySpan<int>) | Calculates an output tensor by slicing the input tensor along given axes with given starts, ends, and steps. |
Softmax(TensorFloat, TensorFloat, int) | Computes an output tensor by applying the |
Softplus(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Softsign(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
SpaceToDepth(TensorFloat, TensorFloat, int) | Computes the output tensor by permuting data from blocks of spatial data into depth. |
Split(Tensor, Tensor, int, int) | Calculates an output tensor by splitting the input tensor along a given axis between start and end. |
Sqrt(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Square(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Sub(TensorFloat, TensorFloat, TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Sub(TensorInt, TensorInt, TensorInt) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Sum(TensorFloat[], TensorFloat) | Performs an element-wise This supports numpy-style broadcasting of input tensors. |
Swish(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Tan(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
Tanh(TensorFloat, TensorFloat) | Computes an output tensor by applying the element-wise |
ThresholdedRelu(TensorFloat, TensorFloat, float) | Computes an output tensor by applying the element-wise |
Tile(Tensor, Tensor, ReadOnlySpan<int>) | Calculates an output tensor by repeating the input layer a given number of times along each axis. |
TopK(TensorFloat, TensorFloat, TensorInt, int, int, bool) | Calculates the top-K largest or smallest elements of an input tensor along a given axis. |
Transpose(Tensor, Tensor) | Calculates an output tensor by reversing the dimensions of the input tensor. |
Transpose(Tensor, Tensor, int[]) | Calculates an output tensor by permuting the axes and data of the input tensor according to the given permutations. |
Tril(Tensor, Tensor, int) | Computes the output tensor by retaining the lower triangular values from an input matrix or matrix batch and setting the other values to zero. |
Triu(Tensor, Tensor, int) | Computes the output tensor by retaining the upper triangular values from an input matrix or matrix batch and setting the other values to zero. |
Where(TensorInt, Tensor, Tensor, Tensor) | Performs an element-wise |
Xor(TensorInt, TensorInt, TensorInt) | Performs an element-wise |