tvm_tirx.tirx.tile.views.BinaryReduce

Contents

tvm_tirx.tirx.tile.views.BinaryReduce#

class tvm_tirx.tirx.tile.views.BinaryReduce#

Bases: TensorCall

Combine a binary operation with a reduction operation.

binary_reduce(binary_output, reduce_output, binary_input1, binary_input2, binary_op, reduce_op, reduce_axes, )

kind = 'binary_reduce'#

binary_output#

reduce_output#

binary_input1#

binary_input2#

binary_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, Any]