tvm_tirx.tirx.tile.views.UnaryOpWithScaleBias#

class tvm_tirx.tirx.tile.views.UnaryOpWithScaleBias#

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

Both scale and bias are required operands, applied before the unary operation. output = unary(input * scale + bias)

property srcs: list[Expr]#: Get the source expressions (inputs) of the operator.

get_private_tensors_trn(tensor_dict, sctx)#

Parameters:

Return type:

dict[str, Any]

tvm_tirx.tirx.tile.views.UnaryOpWithScaleBias