tvm_tirx.tirx.script.call_packed

Contents

tvm_tirx.tirx.script.call_packed#

tvm_tirx.tirx.script.call_packed(*args, loc=UNKNOWN_LOC, ty=None)#

Build expression by call an external packed function.

The argument to a packed function can be an Expr or a tensor variable. The argument is the corresponding POD type when Expr is presented.

When the argument is a Var carrying TensorType, the corresponding PackedFunc will receive an TVMArrayHandle whose content is valid during the callback period. If the PackedFunc is a python callback, then the corresponding argument is Tensor.

Parameters:

  • args (list of Expr or Var.) – Positional arguments.

  • loc (Location, optional) – The location of this operator in the source code.

Returns:

call (Expr) – The call expression.

See also

te.extern

Create tensor with extern function call.

Contents