Skip to content

cli

torch_to_nnef.nemo_tract.cli

CLI for NeMo→NNEF export.

This module owns the full CLI surface: argument parsing, runtime orchestration (run_export), and the main entry point. The programmatic export API lives in export.py.

add_inspection_args

add_inspection_args(parser: argparse.ArgumentParser) -> None

Register inspection and dry-run arguments.

main

main() -> None

CLI entry point: parse config and run export.

parse_config

parse_config() -> NemoTractConfig

Build and parse CLI args into a NemoTractConfig dataclass.

run_export

run_export(cfg: NemoTractConfig) -> None

Orchestrate a full NeMo→NNEF export from a CLI configuration.

setup_inference_target_from_cli_args

setup_inference_target_from_cli_args(cfg: NemoTractConfig) -> TractNNEF

Build a TractNNEF inference target from CLI configuration.