Comparison Functions

Module: polygraphy.tools.args

class CompareFuncSimpleArgs[source]

Bases: polygraphy.tools.args.base.BaseArgs

Comparison Function: simple: the CompareFunc.simple comparison function.

Depends on:

  • ComparatorCompareArgs

parse_impl(args)[source]

Parses command-line arguments and populates the following attributes:

no_shape_check

Whether to skip shape checks.

Type

bool

rtol

Per-tensor relative tolerance.

Type

Dict[str, float]

atol

Per-tensor absolute tolerance.

Type

Dict[str, float]

check_error_stat

The error metric to check.

Type

str

class CompareFuncIndicesArgs[source]

Bases: polygraphy.tools.args.base.BaseArgs

Comparison Function: indices: the CompareFunc.indices comparison function.

Depends on:

  • ComparatorCompareArgs

parse_impl(args)[source]

Parses command-line arguments and populates the following attributes:

index_tolerance

Per-tensor index tolerance.

Type

Dict[str, int]