tvm_tirx.tirx.script.index_map

Contents

tvm_tirx.tirx.script.index_map#

tvm_tirx.tirx.script.index_map(mapping, *, inverse_index_map=None, index_dtype='int64')#

Create a TIR Index mapping

Parameters:

  • mapping (Callable)

  • inverse_index_map (Callable | None)

  • index_dtype (str)

Return type:

IndexMap

Contents