afe.backends.mpk.operator ========================= .. py:module:: afe.backends.mpk.operator .. autoapi-nested-parse:: 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 ------- .. autoapisummary:: afe.backends.mpk.operator.PluginMPKOperatorData Functions --------- .. autoapisummary:: afe.backends.mpk.operator.make_ev_plugin_parameters Module Contents --------------- .. py:class:: PluginMPKOperatorData Fields of PluginMPKData that are produced in an operator-specific way. These fields should be copied into a PluginMPKData to make a plugin. .. py:attribute:: processor :type: str .. py:attribute:: config_params :type: afe.backends.mpk.defines.ConfigParamsMPKData .. py:attribute:: input_nodes :type: List[afe.backends.mpk.defines.PluginInputNodeMPKData] .. py:attribute:: output_nodes :type: List[afe.backends.mpk.defines.InOutNodesMPKData] .. py:function:: 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.