afe.tvm_converter.custom_convert_maps

TODO(Joey.Chou):
Use registration method to register

  • custom_convert_map

  • MergeComposite Pattern _PATTERN_TABLE

  • tvm_func_to_awesome_attributes

Attributes

AVAILABLE_ANNOTATIONS

CUSTOM_CONVERT_MAP_DICT

TVM_FUNC_TO_AWESOME_ATTRIBUTES_DICT

Classes

GroupConv2DTranspose

Models: CenterNet

ConstantMultiplyAdd

Merge multiply with constant to add op.

RenamerWithAttrsChecker

Module Contents

class afe.tvm_converter.custom_convert_maps.GroupConv2DTranspose

Models: CenterNet Author: Alicja Kwasniewska, Joey Chou

Decompose a nn.conv2d_transpose with groups > 1 into splits, multiple nn.conv2d_transpose, and concatenate as below:


split

β€”β€”β€”β€”β€”β€”β€”β€”β€” …. | | | | | …

tuple_get_item …..

nn.conv2d_transpose

| | | | …
β€”β€”β€”β€”β€”β€”β€”β€”β€” ….

tuple

concatenate

ONNX_op_name = 'ConvTranspose'
static map_impl()
static tvm_func_to_awesome_attributes(params: afe.tvm_converter.parameters.TVMConverterParams, func: afe._tvm._defines.TVMFunction) → afe.ir.attributes.AwesomeAttributes

Functions to extract decomposed group conv2d_transpose Relay attributes and translate it into SiMa IR Conv2DTransposeAttrs

class afe.tvm_converter.custom_convert_maps.ConstantMultiplyAdd

Merge multiply with constant to add op. Do it for both inputs if possible.

static tvm_func_to_awesome_attributes(params: afe.tvm_converter.parameters.TVMConverterParams, func: afe._tvm._defines.TVMFunction) → afe.ir.attributes.AwesomeAttributes

Functions to extract decomposed constant, mul, add relay attributes from composite constant_multiply_add and translate it into SiMa IR ConstantMultiplyAddAttrs.

class afe.tvm_converter.custom_convert_maps.RenamerWithAttrsChecker(new_name, checker)
checker
afe.tvm_converter.custom_convert_maps.AVAILABLE_ANNOTATIONS = ['int8', 'int16']
afe.tvm_converter.custom_convert_maps.CUSTOM_CONVERT_MAP_DICT
afe.tvm_converter.custom_convert_maps.TVM_FUNC_TO_AWESOME_ATTRIBUTES_DICT