matx.runtime.container module

Runtime container structures.

class matx.runtime.container.Array(seq=())

Bases: Object

Array container of MATX.

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.

__init__(seq=())

class matx.runtime.container.ArrayIterator(obj)

Bases: object

__init__(obj)

class matx.runtime.container.Map(seq=())

Bases: Object

Map container of MATX IR.

You do not need to create Map explicitly. Normally python dict will be converted automaticall to Map during tvm function call. You can use convert to create a dict[Object-> Object] into a Map

__init__(seq=())

get(key, default=None)

Get an element with a default value.

Parameters:

• key (object) – The attribute key.

• default (object) – The default object.

Returns:

value – The result value.

Return type:

object

items()

Get the items from the map

keys()

Get the keys from the map

values()

Get the values from the map

class matx.runtime.container.MapIterator(obj)

Bases: object

__init__(obj)

class matx.runtime.container.Tuple(*fields)

Bases: Object

tuple object.

Parameters:

fields (List[Object]) – The source.

__init__(*fields)

matx.runtime.container.getitem_helper(obj, elem_getter, length, idx)

Helper function to implement a pythonic getitem function.

Parameters:

• obj (object) – The original object

• elem_getter (function) – A simple function that takes index and return a single element.

• length (int) – The size of the array

• idx (int or slice) – The argument passed to getitem

Returns:

result – The result of getitem

Return type:

object

matx.runtime.container.slice_index_correction(index, length)