Runner Selection

Module: polygraphy.tools.args

class RunnerSelectArgs

Bases: BaseArgs

Runners: selecting runners to use for inference.

add_to_script_impl(script)

Adds all selected runners to the script.

Returns:

The name of the list of runners in the script.

Return type:

str

parse_impl(args)

Parses command-line arguments and populates the following attributes:

runners

A list of tuples mapping runner option strings to human readable names for all selected runners, in the order they were selected in. For example:

[("trt", "TensorRT"), ("onnxrt", "ONNX-Runtime")]

Type:

List[Tuple[str, str]]