tvm_tirx.tirx.script.stack_make_shape

tvm_tirx.tirx.script.stack_make_shape#

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

Allocate a shape tuple on stack, return the handle

Parameters:
  • args (int) – The tuple shape.

  • loc (Location)

Returns:

call (Expr) – The call expression.