tvm_tirx.tirx.script.stack_alloca

Contents

tvm_tirx.tirx.script.stack_alloca#

tvm_tirx.tirx.script.stack_alloca(dtype_str, num, *, ty=None, loc=UNKNOWN_LOC)#

Return new on stack dtype[num]

Parameters:

  • dtype_str (str) – The data type of array.

  • num (int) – The size of array.

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents