polygraphy.tools.args.trt_legacy.TrtLegacyArgs Class Reference

Inheritance diagram for polygraphy.tools.args.trt_legacy.TrtLegacyArgs:

Collaboration diagram for polygraphy.tools.args.trt_legacy.TrtLegacyArgs:

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

Public Attributes
	onnx_loader_args
	model_args
	trt_loader_args
	tf_loader_args
	trt_runner_args
	trt_outputs
	caffe_model
	batch_size
	save_uff
	uff_order
	preprocessor

Member Function Documentation

add_to_parser()

def polygraphy.tools.args.trt_legacy.TrtLegacyArgs.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.

register()

def polygraphy.tools.args.trt_legacy.TrtLegacyArgs.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 from polygraphy.tools.args.base.BaseArgs.

check_registered()

def polygraphy.tools.args.trt_legacy.TrtLegacyArgs.check_registered

(

self

)

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

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

parse()

def polygraphy.tools.args.trt_legacy.TrtLegacyArgs.parse	(		self,
			args
	)

Parses relevant arguments from command-line arguments.

Args:
    args: Arguments provided by argparse.

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

add_to_script()

def polygraphy.tools.args.trt_legacy.TrtLegacyArgs.add_to_script	(		self,
			script
	)

Member Data Documentation

onnx_loader_args

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.onnx_loader_args

model_args

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.model_args

trt_loader_args

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.trt_loader_args

tf_loader_args

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.tf_loader_args

trt_runner_args

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.trt_runner_args

trt_outputs

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.trt_outputs

caffe_model

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.caffe_model

batch_size

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.batch_size

save_uff

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.save_uff

uff_order

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.uff_order

preprocessor

polygraphy.tools.args.trt_legacy.TrtLegacyArgs.preprocessor

---

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

• tools/args/trt_legacy.py