| 181 | |
| 182 | |
| 183 | class _Profiler(trt.IProfiler): |
| 184 | |
| 185 | def __init__(self): |
| 186 | super().__init__() |
| 187 | self.results = [] |
| 188 | |
| 189 | def report_layer_time(self, layer_name, ms): |
| 190 | self.results.append((layer_name, ms)) |
| 191 | |
| 192 | |
| 193 | def _contiguous_tile_beam_width(tensor: torch.Tensor, size: int, |