tvm_tirx.tirx.tile.views.ReduceOp

class tvm_tirx.tirx.tile.views.ReduceOp

Bases: TensorCall

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

Reduction operators reduce one or more dimensions of the input tensor.

input

output

reduce_axes

accum

property srcs: list[Expr]

Get the source expression (input) of the operator.

property dsts: list[Expr]

Get the destination expression (output) 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, Any]