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[source]

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'[source]
static map_impl()[source]
static tvm_func_to_awesome_attributes(params: afe.tvm_converter.parameters.TVMConverterParams, func: afe._tvm._defines.TVMFunction) → afe.ir.attributes.AwesomeAttributes[source]

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

class afe.tvm_converter.custom_convert_maps.ConstantMultiplyAdd[source]

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[source]

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)[source]
checker[source]
afe.tvm_converter.custom_convert_maps.AVAILABLE_ANNOTATIONS = ['int8', 'int16'][source]
afe.tvm_converter.custom_convert_maps.CUSTOM_CONVERT_MAP_DICT[source]
afe.tvm_converter.custom_convert_maps.TVM_FUNC_TO_AWESOME_ATTRIBUTES_DICT[source]