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

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

config_params: afe.backends.mpk.defines.ConfigParamsMPKData[source]

input_nodes: List[afe.backends.mpk.defines.PluginInputNodeMPKData][source]

output_nodes: List[afe.backends.mpk.defines.InOutNodesMPKData][source]

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[source]: Do the operator-specific part of extracting a node’s parameters for an EV plugin.