polygraphy.tools.args.model.ModelArgs Class Reference

Inheritance diagram for polygraphy.tools.args.model.ModelArgs:

Collaboration diagram for polygraphy.tools.args.model.ModelArgs:

Public Member Functions
def	__init__ (self, model_required=False, inputs="--inputs", model_type=None)
def	add_to_parser (self, parser)
def	parse (self, args)
def	register (self, maker)
def	check_registered (self)

Public Attributes
	input_shapes
	model_file
	model_type

Private Attributes
	_model_required
	_inputs
	_model_type

Constructor & Destructor Documentation

__init__()

def polygraphy.tools.args.model.ModelArgs.__init__	(		self,
			model_required = False,
			inputs = "--inputs",
			model_type = None
	)

Member Function Documentation

add_to_parser()

def polygraphy.tools.args.model.ModelArgs.add_to_parser	(		self,
			parser
	)

Add arguments to a command-line parser.

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

Reimplemented from polygraphy.tools.args.base.BaseArgs.

parse()

def polygraphy.tools.args.model.ModelArgs.parse	(		self,
			args
	)

Parses relevant arguments from command-line arguments.

Args:
    args: Arguments provided by argparse.

Reimplemented from polygraphy.tools.args.base.BaseArgs.

register()

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

inherited

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 )

inherited

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.

Member Data Documentation

_model_required

polygraphy.tools.args.model.ModelArgs._model_required

private

_inputs

polygraphy.tools.args.model.ModelArgs._inputs

private

_model_type

polygraphy.tools.args.model.ModelArgs._model_type

private

input_shapes

polygraphy.tools.args.model.ModelArgs.input_shapes

model_file

polygraphy.tools.args.model.ModelArgs.model_file

model_type

polygraphy.tools.args.model.ModelArgs.model_type

---

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

• tools/Polygraphy/polygraphy/tools/args/model.py