MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / BoundedShape

Class BoundedShape

tools/Polygraphy/polygraphy/common/struct.py:25–36  ·  view source on GitHub ↗

Represents a shape with min/max bounds.

Source from the content-addressed store, hash-verified

23
24
25class BoundedShape(list):
26 """
27 Represents a shape with min/max bounds.
28 """
29
30 def __init__(self, shape, min=None, max=None):
31 super().__init__(shape)
32 self.min = min
33 self.max = max
34
35 def __repr__(self):
36 return f"BoundedShape({list(self)}, min={self.min}, max={self.max})"
37
38
39class MetadataTuple:

Callers 3

check_calibratorMethod · 0.90
addMethod · 0.85

Calls

no outgoing calls

Tested by 2

check_calibratorMethod · 0.72