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