Function tvm::script::ir_builder::ir::Grid

Function tvm::script::ir_builder::ir::Grid#

Function Documentation#

ForFrame tvm::script::ir_builder::ir::Grid(ffi::Array<ffi::Variant<PrimExpr, ffi::Tuple<PrimExpr, PrimExpr>>> extents, ffi::Optional<PrimType> dtype = std::nullopt)#

The grid For statement.

Parameters:

  • extents – The extents of the iteration.

  • dtype – The optional dtype of every loop var (“int32” or “uint32”). When omitted each loop var takes the dtype of its own extent.

Returns:

The ForFrame.