(self, parser)
| 301 | return network_results |
| 302 | |
| 303 | def add_args(self, parser) -> argparse.ArgumentParser: |
| 304 | super().add_args(parser) |
| 305 | device_group = parser.add_argument_group("device") |
| 306 | device_group.add_argument( |
| 307 | "--cpu", |
| 308 | help="Run inference using CPU for frameworks.", |
| 309 | action="store_true", |
| 310 | ) |
| 311 | |
| 312 | class TRTInferenceCommand(NetworkCommand): |
| 313 | """Base class that is associated with Polygraphy related scripts.""" |