heterocl.tvm.container module

Container data structures used in TVM DSL.

class Array(handle)

Bases: heterocl.tvm._ffi.node.NodeBase

Array container of TVM.

You do not need to create Array explicitly. Normally python list and tuple will be converted automatically to Array during tvm function call. You may get Array in return values of TVM function call.

class LoweredFunc(handle)

Bases: heterocl.tvm._ffi.node.NodeBase

Represent a LoweredFunc in TVM.

DeviceFunc = 2

HostFunc = 1

MixedFunc = 0

class Map(handle)

Bases: heterocl.tvm._ffi.node.NodeBase

Map container of TVM.

You do not need to create Map explicitly. Normally python dict will be converted automatically to Array during tvm function call. You may get Map in return values of TVM function call.

items()

Get the items from the map

class Range(handle)

Bases: heterocl.tvm._ffi.node.NodeBase

Represent range in TVM.

You do not need to create Range explicitly. Python list and tuple will be converted automatically to Range in api functions.