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.backend.trt.loader.ShapeTuple Class Reference
Inheritance diagram for polygraphy.backend.trt.loader.ShapeTuple:
Collaboration diagram for polygraphy.backend.trt.loader.ShapeTuple:

Public Member Functions

def __init__ (self, min, opt, max)
 
def __str__ (self)
 
def __repr__ (self)
 

Public Attributes

 min
 
 opt
 
 max
 

Constructor & Destructor Documentation

◆ __init__()

def polygraphy.backend.trt.loader.ShapeTuple.__init__ (   self,
  min,
  opt,
  max 
)
Represents a set of shapes for a single binding in a profile

Args:
    min (Tuple[int]): The minimum shape that the profile will support.
    opt (Tuple[int]): The shape for which TensorRT will optimize the engine.
    max (Tuple[int]): The maximum shape that the profile will support.

Member Function Documentation

◆ __str__()

def polygraphy.backend.trt.loader.ShapeTuple.__str__ (   self)
Here is the caller graph for this function:

◆ __repr__()

def polygraphy.backend.trt.loader.ShapeTuple.__repr__ (   self)
Here is the call graph for this function:

Member Data Documentation

◆ min

polygraphy.backend.trt.loader.ShapeTuple.min

◆ opt

polygraphy.backend.trt.loader.ShapeTuple.opt

◆ max

polygraphy.backend.trt.loader.ShapeTuple.max

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