Command-line Argument Groups
Command-line argument groups bundle arguments and related functionality together into reusable components and are used throughout the Polygraphy command-line toolkit.
Argument groups are used to add new command-line tools to Polygraphy or extend existing tools with new functionality.
Module: polygraphy.tools.args
- Base Interface
- Backends
- ONNX
- ONNX-Runtime
- Plugin Reference
- TensorFlow
- TensorRT
- Loaders
TrtLoadPluginsArgsTrtOnnxFlagArgsTrtLoadNetworkArgsTrtLoadNetworkArgs.parse_impl()TrtLoadNetworkArgs.outputsTrtLoadNetworkArgs.exclude_outputsTrtLoadNetworkArgs.trt_network_func_nameTrtLoadNetworkArgs.layer_precisionsTrtLoadNetworkArgs.tensor_datatypesTrtLoadNetworkArgs.tensor_formatsTrtLoadNetworkArgs.postprocess_scriptsTrtLoadNetworkArgs.strongly_typedTrtLoadNetworkArgs.mark_debugTrtLoadNetworkArgs.load_network()
TrtSaveEngineBytesArgsTrtSaveEngineArgsTrtLoadEngineBytesArgsTrtLoadEngineArgsTrtConfigArgsTrtConfigArgs.parse_impl()TrtConfigArgs.profile_dictsTrtConfigArgs.tf32TrtConfigArgs.fp16TrtConfigArgs.bf16TrtConfigArgs.fp8TrtConfigArgs.int8TrtConfigArgs.precision_constraintsTrtConfigArgs.restrictedTrtConfigArgs.calibration_cacheTrtConfigArgs.calibration_base_classTrtConfigArgs.sparse_weightsTrtConfigArgs.load_timing_cacheTrtConfigArgs.load_tacticsTrtConfigArgs.save_tacticsTrtConfigArgs.tactic_sourcesTrtConfigArgs.trt_config_scriptTrtConfigArgs.trt_config_func_nameTrtConfigArgs.trt_config_postprocess_scriptTrtConfigArgs.trt_config_postprocess_func_nameTrtConfigArgs.use_dlaTrtConfigArgs.allow_gpu_fallbackTrtConfigArgs.memory_pool_limitsTrtConfigArgs.engine_capabilityTrtConfigArgs.direct_ioTrtConfigArgs.preview_featuresTrtConfigArgs.refittableTrtConfigArgs.strip_planTrtConfigArgs.builder_optimization_levelTrtConfigArgs.hardware_compatibility_levelTrtConfigArgs.profiling_verbosityTrtConfigArgs.max_aux_streamsTrtConfigArgs.version_compatibleTrtConfigArgs.exclude_lean_runtimeTrtConfigArgs.quantization_flagsTrtConfigArgs.error_on_timing_cache_missTrtConfigArgs.disable_compilation_cacheTrtConfigArgs.weight_streamingTrtConfigArgs.runtime_platformTrtConfigArgs.create_config()
- Runners
- Loaders
- Comparator
- Comparator
- Comparison Functions
CompareFuncSimpleArgsCompareFuncSimpleArgs.parse_impl()CompareFuncSimpleArgs.no_shape_checkCompareFuncSimpleArgs.rtolCompareFuncSimpleArgs.atolCompareFuncSimpleArgs.check_error_statCompareFuncSimpleArgs.infinities_compare_equalCompareFuncSimpleArgs.save_heatmapsCompareFuncSimpleArgs.show_heatmapsCompareFuncSimpleArgs.save_error_metrics_plotCompareFuncSimpleArgs.show_error_metrics_plotCompareFuncSimpleArgs.error_quantile
CompareFuncIndicesArgs
- Data Loader
DataLoaderArgsDataLoaderArgs.parse_impl()DataLoaderArgs.seedDataLoaderArgs.val_rangeDataLoaderArgs.iterationsDataLoaderArgs.load_inputs_pathsDataLoaderArgs.data_loader_scriptDataLoaderArgs.data_loader_func_nameDataLoaderArgs.data_loader_backend_moduleDataLoaderArgs.add_to_script_impl()DataLoaderArgs.get_data_loader()DataLoaderArgs.is_using_random_data()
- Postprocessing Functions
- Logger
- Model