mlonmcu.flow.tflm package

Subpackages

• mlonmcu.flow.tflm.backend package
  • Submodules
  • mlonmcu.flow.tflm.backend.backend module
    • TFLMBackend
      • TFLMBackend.DEFAULTS
      • TFLMBackend.FEATURES
      • TFLMBackend.REQUIRED
      • TFLMBackend.load_model()
      • TFLMBackend.name
      • TFLMBackend.registry
  • mlonmcu.flow.tflm.backend.tflmc module
    • TFLMCBackend
      • TFLMCBackend.DEFAULTS
      • TFLMCBackend.FEATURES
      • TFLMCBackend.REQUIRED
      • TFLMCBackend.generate()
      • TFLMCBackend.generate_header()
      • TFLMCBackend.name
      • TFLMCBackend.print_outputs
  • mlonmcu.flow.tflm.backend.tflmi module
    • TFLMIBackend
      • TFLMIBackend.DEFAULTS
      • TFLMIBackend.FEATURES
      • TFLMIBackend.arena_size
      • TFLMIBackend.custom_ops
      • TFLMIBackend.debug_arena
      • TFLMIBackend.generate()
      • TFLMIBackend.legacy
      • TFLMIBackend.name
      • TFLMIBackend.ops
      • TFLMIBackend.ops_resolver
      • TFLMIBackend.registrations
      • TFLMIBackend.reporter
    • TFLMICodegen
      • TFLMICodegen.generate_header()
      • TFLMICodegen.generate_wrapper()
      • TFLMICodegen.makeCustomOpPrototypes()
      • TFLMICodegen.make_op_registrations()
    • make_hex_array()
  • Module contents
    • TFLMBackend
      • TFLMBackend.DEFAULTS
      • TFLMBackend.FEATURES
      • TFLMBackend.REQUIRED
      • TFLMBackend.load_model()
      • TFLMBackend.name
      • TFLMBackend.registry
    • TFLMCBackend
      • TFLMCBackend.DEFAULTS
      • TFLMCBackend.FEATURES
      • TFLMCBackend.REQUIRED
      • TFLMCBackend.generate()
      • TFLMCBackend.generate_header()
      • TFLMCBackend.name
      • TFLMCBackend.print_outputs
    • TFLMIBackend
      • TFLMIBackend.DEFAULTS
      • TFLMIBackend.FEATURES
      • TFLMIBackend.arena_size
      • TFLMIBackend.custom_ops
      • TFLMIBackend.debug_arena
      • TFLMIBackend.generate()
      • TFLMIBackend.legacy
      • TFLMIBackend.name
      • TFLMIBackend.ops
      • TFLMIBackend.ops_resolver
      • TFLMIBackend.registrations
      • TFLMIBackend.reporter

Submodules

mlonmcu.flow.tflm.framework module

Definitions for TFLMFramework.

class mlonmcu.flow.tflm.framework.TFLMFramework(features=None, config=None)

Bases: Framework

TFLM Framework specialization.

DEFAULTS = {'optimized_kernel': None, 'optimized_kernel_inc_dirs': [], 'optimized_kernel_libs': []}

FEATURES = {'cmsisnn', 'muriscvnn'}

REQUIRED = {'tf.src_dir'}

backends = {'tflmc': <class 'mlonmcu.flow.tflm.backend.tflmc.TFLMCBackend'>, 'tflmi': <class 'mlonmcu.flow.tflm.backend.tflmi.TFLMIBackend'>}

get_platform_defs(platform)

name = 'tflm'

property optimized_kernel

property optimized_kernel_inc_dirs

property optimized_kernel_libs

property tf_src

Module contents

TFLite framework module.

class mlonmcu.flow.tflm.TFLMBackend(features=None, config=None)

Bases: Backend

DEFAULTS = {}

FEATURES = {}

REQUIRED = {}

load_model(model, input_shapes=None, output_shapes=None, input_types=None, output_types=None)

name = None

registry = {'tflmc': <class 'mlonmcu.flow.tflm.backend.tflmc.TFLMCBackend'>, 'tflmi': <class 'mlonmcu.flow.tflm.backend.tflmi.TFLMIBackend'>}

class mlonmcu.flow.tflm.TFLMCBackend(features=None, config=None)

Bases: TFLMBackend

DEFAULTS = {'custom_ops': [], 'debug_arena': False, 'print_outputs': False, 'registrations': {}}

FEATURES = {'debug_arena'}

REQUIRED = {'tflmc.exe'}

generate() → Tuple[dict, dict]

generate_header()

name = 'tflmc'

property print_outputs

class mlonmcu.flow.tflm.TFLMIBackend(features=None, config=None)

Bases: TFLMBackend

DEFAULTS = {'arena_size': 1048576, 'custom_ops': [], 'debug_arena': False, 'legacy': False, 'ops': [], 'ops_resolver': 'mutable', 'registrations': {}, 'reporter': False}

FEATURES = {'debug_arena'}

property arena_size

property custom_ops

property debug_arena

generate() → Tuple[dict, dict]

property legacy

name = 'tflmi'

property ops

property ops_resolver

property registrations

property reporter