polygraphy.tools.args.base.BaseArgs Class Reference

Inheritance diagram for polygraphy.tools.args.base.BaseArgs:

Collaboration diagram for polygraphy.tools.args.base.BaseArgs:

Public Member Functions
def	add_to_parser (self, parser)
def	parse (self, args)
def	register (self, maker)
def	check_registered (self)

Detailed Description

Adds a arguments to a command-line parser, and provides capabilities to create
Polygraphy objects based on the arguments.

Member Function Documentation

add_to_parser()

def polygraphy.tools.args.base.BaseArgs.add_to_parser	(		self,
			parser
	)

Add arguments to a command-line parser.

Args:
    parser (argparse.ArgumentParser): The argument parser.

Reimplemented in polygraphy.tools.args.comparator.ComparatorCompareArgs, polygraphy.tools.args.model.ModelArgs, polygraphy.tools.args.trt.loader.TrtLoaderArgs, polygraphy.tools.args.onnx.loader.OnnxLoaderArgs, polygraphy.tools.args.comparator.ComparatorRunArgs, polygraphy.tools.args.tf.loader.TfLoaderArgs, polygraphy.tools.args.data_loader.DataLoaderArgs, polygraphy.tools.args.trt.runner.TrtRunnerArgs, polygraphy.tools.args.trt_legacy.TrtLegacyArgs, polygraphy.tools.args.logger.LoggerArgs, polygraphy.tools.args.tf2onnx.loader.Tf2OnnxLoaderArgs, polygraphy.tools.args.tf.config.TfConfigArgs, and polygraphy.tools.args.tf.runner.TfRunnerArgs.

parse()

def polygraphy.tools.args.base.BaseArgs.parse	(		self,
			args
	)

Parses relevant arguments from command-line arguments.

Args:
    args: Arguments provided by argparse.

Reimplemented in polygraphy.tools.args.comparator.ComparatorCompareArgs, polygraphy.tools.args.trt.loader.TrtLoaderArgs, polygraphy.tools.args.tf.loader.TfLoaderArgs, polygraphy.tools.args.onnx.loader.OnnxLoaderArgs, polygraphy.tools.args.trt_legacy.TrtLegacyArgs, polygraphy.tools.args.data_loader.DataLoaderArgs, polygraphy.tools.args.trt.runner.TrtRunnerArgs, polygraphy.tools.args.model.ModelArgs, polygraphy.tools.args.comparator.ComparatorRunArgs, polygraphy.tools.args.tf.runner.TfRunnerArgs, polygraphy.tools.args.tf2onnx.loader.Tf2OnnxLoaderArgs, polygraphy.tools.args.logger.LoggerArgs, and polygraphy.tools.args.tf.config.TfConfigArgs.

register()

def polygraphy.tools.args.base.BaseArgs.register	(		self,
			maker
	)

Registers another argument group with this one.
This can be used to pick up dependencies for example.

Args:
    maker (BaseArgs): Another argument group.

Reimplemented in polygraphy.tools.args.trt.loader.TrtLoaderArgs, polygraphy.tools.args.tf.loader.TfLoaderArgs, polygraphy.tools.args.onnx.loader.OnnxLoaderArgs, polygraphy.tools.args.data_loader.DataLoaderArgs, polygraphy.tools.args.trt_legacy.TrtLegacyArgs, polygraphy.tools.args.trt.runner.TrtRunnerArgs, polygraphy.tools.args.tf2onnx.loader.Tf2OnnxLoaderArgs, polygraphy.tools.args.tf.runner.TfRunnerArgs, polygraphy.tools.args.onnx.runner.OnnxtfRunnerArgs, and polygraphy.tools.args.onnxrt.runner.OnnxrtRunnerArgs.

check_registered()

def polygraphy.tools.args.base.BaseArgs.check_registered

(

self

)

Called after all `register()` calls to make dependency checks easier.

Reimplemented in polygraphy.tools.args.trt.loader.TrtLoaderArgs, polygraphy.tools.args.tf.loader.TfLoaderArgs, polygraphy.tools.args.onnx.loader.OnnxLoaderArgs, polygraphy.tools.args.trt_legacy.TrtLegacyArgs, polygraphy.tools.args.data_loader.DataLoaderArgs, polygraphy.tools.args.trt.runner.TrtRunnerArgs, polygraphy.tools.args.tf.runner.TfRunnerArgs, polygraphy.tools.args.tf2onnx.loader.Tf2OnnxLoaderArgs, polygraphy.tools.args.onnx.runner.OnnxtfRunnerArgs, and polygraphy.tools.args.onnxrt.runner.OnnxrtRunnerArgs.

---

The documentation for this class was generated from the following file:

• args/base.py