afe.backends.mpk.operator

Operator-specific conversion from SiMaIR operators to plugin sections in a MPK JSON file. An operator can be converted to MPK JSON data by calling make_ev_plugin_parameters. The result should be combined with operator-independent data to make a PluginMPKData.

Classes

PluginMPKOperatorData

Fields of PluginMPKData that are produced in an operator-specific way.

Functions

make_ev_plugin_parameters(→ PluginMPKOperatorData)

Do the operator-specific part of extracting a node's parameters for an EV plugin.

Module Contents

class afe.backends.mpk.operator.PluginMPKOperatorData

Fields of PluginMPKData that are produced in an operator-specific way. These fields should be copied into a PluginMPKData to make a plugin.

processor: str

config_params: afe.backends.mpk.defines.ConfigParamsMPKData

input_nodes: list[afe.backends.mpk.defines.PluginInputNodeMPKData]

output_nodes: list[afe.backends.mpk.defines.InOutNodesMPKData]

afe.backends.mpk.operator.make_ev_plugin_parameters(node: afe.ir.node.AwesomeNode, input_nodes: list[afe.backends.mpk.defines.PluginInputNodeMPKData], desired_batch_size: int, actual_batch_size: int, output_names: dict[str, str]) → PluginMPKOperatorData: Do the operator-specific part of extracting a node’s parameters for an EV plugin.