Expand description
tosa dialect.
The Tensor Operator Set Architecture (TOSA) dialect.
This dialect implements the TOSA standard described at https://www.mlplatform.org/tosa/tosa_spec.html .
Tensor Operator Set Architecture (TOSA) provides a set of whole-tensor operations commonly employed by Deep Neural Networks. The intent is to enable a variety of implementations running on a diverse range of processors, with the results at the TOSA level consistent across those implementations. Applications or frameworks which target TOSA can therefore be deployed on a wide range of different processors, such as CPUs or GPUs, with defined accuracy and compatibility constraints. Most operators from the common ML frameworks should be expressible in TOSA. It is expected that there will be tools to lower from the ML frameworks into TOSA.
Structs§
- An
absoperation. Elementwise abs op. - A builder for an
absoperation. - An
addoperation. Elementwise addition operator. - A builder for an
addoperation. - An
apply_scaleoperation. Rescale scalar operator for Tosa tensor operators. - A builder for an
apply_scaleoperation. - An
argmaxoperation. Perform argmax on the input.. - A builder for an
argmaxoperation. - An
arithmetic_right_shiftoperation. Elementwise Arithmetic Right Shift. - A builder for an
arithmetic_right_shiftoperation. - An
avg_pool2doperation. Performs max pooling on the input.. - A builder for an
avg_pool2doperation. - A
bitwise_andoperation. Bitwise AND operator. - A builder for a
bitwise_andoperation. - A
bitwise_notoperation. Bitwise NOT operator. - A builder for a
bitwise_notoperation. - A
bitwise_oroperation. Bitwise OR operator. - A builder for a
bitwise_oroperation. - A
bitwise_xoroperation. Bitwise XOR operator. - A builder for a
bitwise_xoroperation. - A
castoperation. Cast operation. - A builder for a
castoperation. - A
ceiloperation. Elementwise ceil op. - A builder for a
ceiloperation. - A
clampoperation. Computes clamp(features, min, max).. - A builder for a
clampoperation. - A
clzoperation. Elementwise count leading zero op. - A builder for a
clzoperation. - A
concatoperation. Concatenates tensors along one dimension.. - A builder for a
concatoperation. - A
constoperation. Constant op.. - A builder for a
constoperation. - A
conv2doperation. 2D Convolution Operator. - A builder for a
conv2doperation. - A
conv3doperation. 3D Convolution operator. - A builder for a
conv3doperation. - A
customoperation. Custom operator wrapper for Tosa. - A builder for a
customoperation. - A
depthwise_conv2doperation. Depthwise 2D Convolution operator. - A builder for a
depthwise_conv2doperation. - A
divoperation. Integer divide operator. - A builder for a
divoperation. - An
equaloperation. Returns the truth value of (x == y) element-wise.. - A builder for an
equaloperation. - An
erfoperation. Computes gauss error function of input. - A builder for an
erfoperation. - An
expoperation. Elementwise exp op. - A builder for an
expoperation. - A
fft2doperation. Performs FFT2D operation on the input.. - A builder for a
fft2doperation. - A
flooroperation. Elementwise floor op. - A builder for a
flooroperation. - A
fully_connectedoperation. Fully Connected operator. - A builder for a
fully_connectedoperation. - A
gatheroperation. Gather operation,. - A builder for a
gatheroperation. - A
greater_equaloperation. Returns the truth value of (x >= y) element-wise.. - A builder for a
greater_equaloperation. - A
greateroperation. Returns the truth value of (x > y) element-wise.. - A builder for a
greateroperation. - An
identityoperation. Identity operator. - A builder for an
identityoperation. - A
cond_ifoperation. Conditional if operator. - A builder for a
cond_ifoperation. - A
logoperation. Elementwise log op. - A builder for a
logoperation. - A
logical_andoperation. Returns the truth value of x AND y element-wise.. - A builder for a
logical_andoperation. - A
logical_left_shiftoperation. Elementwise Logical Left Shift. - A builder for a
logical_left_shiftoperation. - A
logical_notoperation. Returns the truth value of NOT x element-wise.. - A builder for a
logical_notoperation. - A
logical_oroperation. Returns the truth value of x OR y element-wise.. - A builder for a
logical_oroperation. - A
logical_right_shiftoperation. Elementwise Logical Right Shift. - A builder for a
logical_right_shiftoperation. - A
logical_xoroperation. Returns the truth value of x XOR y element-wise.. - A builder for a
logical_xoroperation. - A
matmuloperation. Matrix multiplication with bias. - A builder for a
matmuloperation. - A
max_pool2doperation. Performs max pooling on the input.. - A builder for a
max_pool2doperation. - A
maximumoperation. Elementwise Maximum. - A builder for a
maximumoperation. - A
minimumoperation. Elementwise Minimum. - A builder for a
minimumoperation. - A
muloperation. Multiplication operator. - A builder for a
muloperation. - A
negateoperation. Elementwise negate op. - A builder for a
negateoperation. - A
padoperation. Pads a tensor with value specified.. - A builder for a
padoperation. - A
powoperation. Computes the power of one value to another.. - A builder for a
powoperation. - A
rfft2doperation. Performs RFFT2D operation on the input.. - A builder for a
rfft2doperation. - A
reciprocaloperation. Elementwise reciprocal op. - A builder for a
reciprocaloperation. - A
reduce_alloperation. Reduce All operator. - A builder for a
reduce_alloperation. - A
reduce_anyoperation. Reduce Any operator. - A builder for a
reduce_anyoperation. - A
reduce_maxoperation. Reduce Max operator. - A builder for a
reduce_maxoperation. - A
reduce_minoperation. Reduce Min operator. - A builder for a
reduce_minoperation. - A
reduce_prodoperation. Reduce Prod operator. - A builder for a
reduce_prodoperation. - A
reduce_sumoperation. Reduce Sum operator. - A builder for a
reduce_sumoperation. - A
rescaleoperation. Tosa rescale operator. - A builder for a
rescaleoperation. - A
reshapeoperation. Reshape operator. - A builder for a
reshapeoperation. - A
resizeoperation. Resize operation, supports various resize/upsample modes. - A builder for a
resizeoperation. - A
reverseoperation. Reverse operator. - A builder for a
reverseoperation. - A
rsqrtoperation. Elementwise 1/sqrt op. - A builder for a
rsqrtoperation. - A
scatteroperation. Scatter operation,. - A builder for a
scatteroperation. - A
selectoperation. Elementwise select operator. - A builder for a
selectoperation. - A
sigmoidoperation. Computes elementwise sigmoid of input.. - A builder for a
sigmoidoperation. - A
sliceoperation. Slice operator. - A builder for a
sliceoperation. - A
suboperation. Elementwise subtraction operator. - A builder for a
suboperation. - A
tableoperation. Table lookup op. - A builder for a
tableoperation. - A
tanhoperation. Computes elementwise hyperbolic tangent of input. - A builder for a
tanhoperation. - A
tileoperation. Tile operator. - A builder for a
tileoperation. - A
transpose_conv2doperation. Transpose 2D Convolution operator.. - A builder for a
transpose_conv2doperation. - A
transposeoperation. Transpose operator. - A builder for a
transposeoperation. - A
variableoperation. Defines a variable. - A builder for a
variableoperation. - A
variable.readoperation. Read_buffer operator. - A builder for a
variable.readoperation. - A
variable.writeoperation. Write_buffer operator. - A builder for a
variable.writeoperation. - A
while_loopoperation. Output = input; While (Cond(output)) {output = Body(output)}. - A builder for a
while_loopoperation. - A
yieldoperation. Yield operator. - A builder for a
yieldoperation.
Functions§
- Creates an
absoperation. - Creates an
addoperation. - Creates an
apply_scaleoperation. - Creates an
argmaxoperation. - Creates an
arithmetic_right_shiftoperation. - Creates an
avg_pool2doperation. - Creates a
bitwise_andoperation. - Creates a
bitwise_notoperation. - Creates a
bitwise_oroperation. - Creates a
bitwise_xoroperation. - Creates a
castoperation. - Creates a
ceiloperation. - Creates a
clampoperation. - Creates a
clzoperation. - Creates a
concatoperation. - Creates a
cond_ifoperation. - Creates a
constoperation. - Creates a
conv2doperation. - Creates a
conv3doperation. - Creates a
customoperation. - Creates a
depthwise_conv2doperation. - Creates a
divoperation. - Creates an
equaloperation. - Creates an
erfoperation. - Creates an
expoperation. - Creates a
fft2doperation. - Creates a
flooroperation. - Creates a
fully_connectedoperation. - Creates a
gatheroperation. - Creates a
greateroperation. - Creates a
greater_equaloperation. - Creates an
identityoperation. - Creates a
logoperation. - Creates a
logical_andoperation. - Creates a
logical_left_shiftoperation. - Creates a
logical_notoperation. - Creates a
logical_oroperation. - Creates a
logical_right_shiftoperation. - Creates a
logical_xoroperation. - Creates a
matmuloperation. - Creates a
max_pool2doperation. - Creates a
maximumoperation. - Creates a
minimumoperation. - Creates a
muloperation. - Creates a
negateoperation. - Creates a
padoperation. - Creates a
powoperation. - Creates a
reciprocaloperation. - Creates a
reduce_alloperation. - Creates a
reduce_anyoperation. - Creates a
reduce_maxoperation. - Creates a
reduce_minoperation. - Creates a
reduce_prodoperation. - Creates a
reduce_sumoperation. - Creates a
rescaleoperation. - Creates a
reshapeoperation. - Creates a
resizeoperation. - Creates a
reverseoperation. - Creates a
rfft2doperation. - Creates a
rsqrtoperation. - Creates a
scatteroperation. - Creates a
selectoperation. - Creates a
sigmoidoperation. - Creates a
sliceoperation. - Creates a
suboperation. - Creates a
tableoperation. - Creates a
tanhoperation. - Creates a
tileoperation. - Creates a
transposeoperation. - Creates a
transpose_conv2doperation. - Creates a
variableoperation. - Creates a
variable.readoperation. - Creates a
variable.writeoperation. - Creates a
while_loopoperation. - Creates a
yieldoperation.