tvm_tirx.tirx.tile.views.FMA

tvm_tirx.tirx.tile.views.FMA#

class tvm_tirx.tirx.tile.views.FMA#

Bases: TensorCall

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

fma(output, input, scale, bias)

scale and bias can each be either a TensorRegion or an Expr scalar.

kind = 'fma'#
output#
input#
scale#
bias#
property srcs: list[Expr]#

Get the source expressions (inputs) of the operator.

property dsts: list[Expr]#

Get the destination expression (output) of the operator.