TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
polygraphy.tools.precision.precision.STWorstFirst Class Reference
Inheritance diagram for polygraphy.tools.precision.precision.STWorstFirst:
Collaboration diagram for polygraphy.tools.precision.precision.STWorstFirst:

Public Member Functions

def __init__ (self)
 
def add_parser_args (self, parser)
 
def find (self)
 
def add_parser_args (self, parser, mode=True)
 
def run (self, args)
 
def layer_indices (self, num_layers)
 
def mark_layers (self, indices)
 
def check_network (self, suffix)
 
def subscribe_args (self, maker)
 
def setup_parser (self, subparsers)
 
def __call__ (self, args)
 

Public Attributes

 precision
 
 args
 
 golden
 
 parser
 
 name
 
 makers
 

Detailed Description

[EXPERIMENTAL] Progressively mark the worst layers (those that introduce the largest error)
to run in a higher precision. Assumes that the layerwise outputs
are topologically sorted.

Constructor & Destructor Documentation

◆ __init__()

def polygraphy.tools.precision.precision.STWorstFirst.__init__ (   self)

Member Function Documentation

◆ add_parser_args() [1/2]

def polygraphy.tools.precision.precision.STWorstFirst.add_parser_args (   self,
  parser 
)

Reimplemented from polygraphy.tools.base.tool.Tool.

◆ find()

def polygraphy.tools.precision.precision.STWorstFirst.find (   self)
Here is the call graph for this function:

◆ add_parser_args() [2/2]

def polygraphy.tools.precision.precision.STCheckerBase.add_parser_args (   self,
  parser,
  mode = True 
)
inherited

◆ run()

def polygraphy.tools.precision.precision.STCheckerBase.run (   self,
  args 
)
inherited

Reimplemented from polygraphy.tools.base.tool.Tool.

Here is the caller graph for this function:

◆ layer_indices()

def polygraphy.tools.precision.precision.STCheckerBase.layer_indices (   self,
  num_layers 
)
inherited
Here is the caller graph for this function:

◆ mark_layers()

def polygraphy.tools.precision.precision.STCheckerBase.mark_layers (   self,
  indices 
)
inherited
Here is the caller graph for this function:

◆ check_network()

def polygraphy.tools.precision.precision.STCheckerBase.check_network (   self,
  suffix 
)
inherited
Checks whether the provided network is accurate compared to golden values.

Returns:
    OrderedDict[str, OutputCompareResult]:
    A mapping of output names to an object describing whether they matched, and what the
    required tolerances were.
Here is the caller graph for this function:

◆ subscribe_args()

def polygraphy.tools.base.tool.Tool.subscribe_args (   self,
  maker 
)
inherited
Here is the caller graph for this function:

◆ setup_parser()

def polygraphy.tools.base.tool.Tool.setup_parser (   self,
  subparsers 
)
inherited

◆ __call__()

def polygraphy.tools.base.tool.Tool.__call__ (   self,
  args 
)
inherited

Reimplemented in polygraphy.tools.registry.MissingTool.

Here is the call graph for this function:

Member Data Documentation

◆ precision

polygraphy.tools.precision.precision.STCheckerBase.precision
inherited

◆ args

polygraphy.tools.precision.precision.STCheckerBase.args
inherited

◆ golden

polygraphy.tools.precision.precision.STCheckerBase.golden
inherited

◆ parser

polygraphy.tools.precision.precision.STCheckerBase.parser
inherited

◆ name

polygraphy.tools.base.tool.Tool.name
inherited

◆ makers

polygraphy.tools.base.tool.Tool.makers
inherited

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