afe.ir.serializer.visualizer

AwesomeNode to Netron conversion.

Attributes

WEIGHT_ATTRS_NAME

BIAS_ATTRS_NAME

netron_name_map

Classes

NetronOptions

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

Functions

create_netron_graph()Β β†’Β Any)

Produce a Netron-readable representation of a network. The return value is a

Module Contents

afe.ir.serializer.visualizer.WEIGHT_ATTRS_NAME = 'weights_attrs'[source]
afe.ir.serializer.visualizer.BIAS_ATTRS_NAME = 'bias_attrs'[source]
afe.ir.serializer.visualizer.netron_name_map[source]
class afe.ir.serializer.visualizer.NetronOptions[source]

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

afe.ir.serializer.visualizer.create_netron_graph(net: afe.ir.net.AwesomeNet, model_name: str, options: NetronOptions = NetronOptions()) → Any[source]

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.

Parameters:

  • net – Network to convert to JSON

  • model_name – Name of the model

  • options – Representation options

Returns:

JSON-representable Python object