afe.ir.serializer.visualizer
============================

.. py:module:: afe.ir.serializer.visualizer

.. autoapi-nested-parse::

   AwesomeNode to Netron conversion.



Attributes
----------

.. autoapisummary::

   afe.ir.serializer.visualizer.WEIGHT_ATTRS_NAME
   afe.ir.serializer.visualizer.BIAS_ATTRS_NAME
   afe.ir.serializer.visualizer.netron_name_map


Classes
-------

.. autoapisummary::

   afe.ir.serializer.visualizer.NetronOptions


Functions
---------

.. autoapisummary::

   afe.ir.serializer.visualizer.create_netron_graph


Module Contents
---------------

.. py:data:: WEIGHT_ATTRS_NAME
   :value: 'weights_attrs'


.. py:data:: BIAS_ATTRS_NAME
   :value: 'bias_attrs'


.. py:data:: netron_name_map

.. py:class:: NetronOptions

   Options controlling how to produce a Netron graph from an AwesomeNet.


.. py:function:: create_netron_graph(net: afe.ir.net.AwesomeNet, model_name: str, options: NetronOptions = NetronOptions()) -> Any

   Produce a Netron-readable representation of a network.  The return value is a
   JSON-representable Python object, which can be read in Netron as a Keras JSON file.

   :param net: Network to convert to JSON
   :param model_name: Name of the model
   :param options: Representation options
   :return: JSON-representable Python object