tvm_tirx.tirx.tile.views

Contents

tvm_tirx.tirx.tile.views#

Typed views of tensor instruction operands during tile lowering.

Classes#

UnaryOp()

Base class for unary operators: unary(output, input).

UnaryOpWithScaleBias()

Extended unary operator with bias and scale parameters: unary_with_scale_bias(output, input, scale, bias).

BinaryOp()

Base class for binary operators: binary(output, input0, input1).

ReduceOp()

Base class for reduction operators: reduce(output, input, reduce_axes, accum).

Zero()

Zero out all elements in src and store to dst.

Sqrt()

Compute sqrt(src) and store to dst.

SqrtWithScaleBias()

Compute sqrt(src * scale + bias) and store to dst.

Fill()

Fill dst with a scalar value.

Add()

Add src1 and src2 element-wise and store to dst.

Sub()

Subtract src2 from src1 element-wise and store to dst.

Mul()

Multiply src1 and src2 element-wise and store to dst.

FDiv()

Divide src1 by src2 element-wise using floating point division and store to dst.

FMA()

Fused multiply-add: output = input * scale + bias.

Cast()

Cast src to dst.

Copy()

Copy all elements from src to dst.

CopyAsync()

Copy all elements from src to dst asynchronously.

Gemm()

General matrix multiplication: D = A * B * alpha + C * beta.

GemmAsync()

General matrix multiplication asynchronously.

Sum()

Sum elements in src along specified axes and store in dst.

Max()

Compute maximum value in src along specified axes and store in dst.

Min()

Compute minimum value in src along specified axes and store in dst.

Reciprocal()

Compute reciprocal (1/x) for all elements in src and store to dst.

SiLU()

Compute SiLU (x * sigmoid(x)) for all elements in src and store to dst.

Memset()

Set all elements in dst to a specified value.

Maximum()

Compute element-wise maximum of src1 and src2 and store to dst.

Minimum()

Compute element-wise minimum of src1 and src2 and store to dst.

Exp()

Compute exp(src) and store to dst.

ExpWithScaleBias()

Compute exp(src * scale + bias) and store to dst.

Exp2()

Compute exp2(src) and store to dst.

Exp2WithScaleBias()

Compute exp2(src * scale + bias) and store to dst.

Log2()

Compute log2(src) and store to dst.

Log2WithScaleBias()

Compute log2(src * scale + bias) and store to dst.

Select()

Select elements from src1 or src2 based on the predicate.

BinaryReduce()

Combine a binary operation with a reduction operation.

UnaryReduce()

Combine a unary operation with a reduction operation.

UnaryReduceWithScaleBias()

Write unary(input * scale + bias), then reduce the unary output.

BinaryChain()

Chain multiple binary operations together.

ReduceNegate()

Negate the result of a reduction operation.
Contents