tvm_tirx.tirx.tile.views.UnaryReduce

class tvm_tirx.tirx.tile.views.UnaryReduce

Bases: TensorCall

Combine a unary operation with a reduction operation.

unary_reduce(unary_output, reduce_output, unary_input, unary_op, reduce_op, reduce_axes)

kind = 'unary_reduce'

unary_output

reduce_output

unary_input

unary_op

reduce_op

reduce_axes

property srcs: list[Expr]

Get the source expressions (inputs) of the operator.

property dsts: list[Expr]

Get the destination expressions (outputs) of the operator.

get_private_tensors_trn(tensor_dict, sctx)

Parameters:

• op (TensorCall)

• tensor_dict (dict[Any, tuple[Bind, Stmt | None]])

• sctx (DispatchContext)

Return type:

dict[str, Var]