mlonmcu.flow package

Subpackages

• mlonmcu.flow.tflm package
  • Subpackages
    • mlonmcu.flow.tflm.backend package
      • Submodules
      • mlonmcu.flow.tflm.backend.backend module
      • mlonmcu.flow.tflm.backend.tflmc module
      • mlonmcu.flow.tflm.backend.tflmi module
      • Module contents
  • Submodules
  • mlonmcu.flow.tflm.framework module
    • TFLMFramework
      • TFLMFramework.DEFAULTS
      • TFLMFramework.FEATURES
      • TFLMFramework.REQUIRED
      • TFLMFramework.backends
      • TFLMFramework.get_platform_defs()
      • TFLMFramework.name
      • TFLMFramework.optimized_kernel
      • TFLMFramework.optimized_kernel_inc_dirs
      • TFLMFramework.optimized_kernel_libs
      • TFLMFramework.tf_src
  • 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
• mlonmcu.flow.tvm package
  • Subpackages
    • mlonmcu.flow.tvm.backend package
      • Submodules
      • mlonmcu.flow.tvm.backend.backend module
      • mlonmcu.flow.tvm.backend.model_info module
      • mlonmcu.flow.tvm.backend.python_utils module
      • mlonmcu.flow.tvm.backend.tuner module
      • mlonmcu.flow.tvm.backend.tvmaot module
      • mlonmcu.flow.tvm.backend.tvmaotplus module
      • mlonmcu.flow.tvm.backend.tvmc_utils module
      • mlonmcu.flow.tvm.backend.tvmcg module
      • mlonmcu.flow.tvm.backend.tvmllvm module
      • mlonmcu.flow.tvm.backend.tvmrt module
      • mlonmcu.flow.tvm.backend.wrapper module
      • Module contents
  • Submodules
  • mlonmcu.flow.tvm.framework module
    • TVMFramework
      • TVMFramework.DEFAULTS
      • TVMFramework.FEATURES
      • TVMFramework.REQUIRED
      • TVMFramework.crt_config_dir
      • TVMFramework.extra_incs
      • TVMFramework.extra_libs
      • TVMFramework.get_platform_defs()
      • TVMFramework.name
      • TVMFramework.tvm_src
    • get_crt_config_dir()
  • Module contents
    • TVMAOTBackend
      • TVMAOTBackend.DEFAULTS
      • TVMAOTBackend.FEATURES
      • TVMAOTBackend.alignment_bytes
      • TVMAOTBackend.arena_size
      • TVMAOTBackend.debug_arena
      • TVMAOTBackend.generate()
      • TVMAOTBackend.get_tvmc_compile_args()
      • TVMAOTBackend.get_workspace_size_from_metadata()
      • TVMAOTBackend.name
      • TVMAOTBackend.unpacked_api
    • TVMAOTPlusBackend
      • TVMAOTPlusBackend.DEFAULTS
      • TVMAOTPlusBackend.name
    • TVMCGBackend
      • TVMCGBackend.REQUIRED
      • TVMCGBackend.generate()
      • TVMCGBackend.get_max_workspace_size_from_metadata()
      • TVMCGBackend.name
    • TVMRTBackend
      • TVMRTBackend.DEFAULTS
      • TVMRTBackend.FEATURES
      • TVMRTBackend.arena_size
      • TVMRTBackend.debug_arena
      • TVMRTBackend.generate()
      • TVMRTBackend.get_tvmc_compile_args()
      • TVMRTBackend.name

Submodules

mlonmcu.flow.backend module

class mlonmcu.flow.backend.Backend(framework='', features=None, config=None)

Bases: ABC

DEFAULTS = {}

FEATURES = {}

OPTIONAL = {}

REQUIRED = {}

add_platform_config(platform, config)

add_platform_defs(platform, defs)

export_artifacts(path)

abstract generate() → Tuple[dict, dict]

generate_artifacts() → List[Artifact]

get_platform_config(platform)

get_platform_defs(platform)

property has_tuner

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

name = None

property needs_target

process_features(features)

set_tuning_records(filepath)

supports_model(model)

mlonmcu.flow.backend.get_parser(backend_name, features, required, defaults)

mlonmcu.flow.backend.init_backend_features(names, config)

mlonmcu.flow.backend.main(backend, args=None)

mlonmcu.flow.framework module

class mlonmcu.flow.framework.Framework(features=None, config=None, backends={})

Bases: ABC

DEFAULTS = {}

FEATURES = {}

OPTIONAL = {}

REQUIRED = {'tf.src_dir'}

add_platform_config(platform, config)

add_platform_defs(platform, defs)

get_platform_config(platform)

get_platform_defs(platform)

name = None

process_features(features)

registry = {'tflm': <class 'mlonmcu.flow.tflm.framework.TFLMFramework'>, 'tvm': <class 'mlonmcu.flow.tvm.framework.TVMFramework'>}

remove_config_prefix(config)

Module contents

Flow module for frameworks and backend.

mlonmcu.flow.get_available_backend_names()

Return all available backend names.