Loaders

Module: polygraphy.tools.args

class OnnxrtSessionArgs[source]

Bases: polygraphy.tools.args.base.BaseArgs

ONNX-Runtime Session Creation: creating an ONNX-Runtime Inference Session

Depends on:

  • OnnxLoadArgs

  • ModelArgs

parse_impl(args)[source]

Parses command-line arguments and populates the following attributes:

providers

A list of execution providers.

Type

List[str]

load_onnxrt_session()[source]

Loads an ONNX-Runtime Inference Session according to arguments provided on the command-line.

Returns

onnxruntime.InferenceSession