Function tvm::tirx::stack_make_dltensor_op

Function tvm::tirx::stack_make_dltensor_op#

Function Documentation#

const Op &tvm::tirx::stack_make_dltensor_op()#

Construct a stack-allocated DLTensor.

Arguments, in order:

  • args[0]: data, The input data.

  • args[1]: shape, The shape.

  • args[2]: strides, The strides.

  • args[3]: ndim, The number of dimensions.

  • args[4]: arr_dtype, The array data type.

  • args[5]: elem_offset, The element offset.