◆ __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.
◆ __str__()
def polygraphy.backend.trt.loader.ShapeTuple.__str__ |
( |
|
self | ) |
|
◆ __repr__()
def polygraphy.backend.trt.loader.ShapeTuple.__repr__ |
( |
|
self | ) |
|
◆ 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: