(self, input_example)
| 400 | os.remove(self.plan_path) |
| 401 | |
| 402 | def _inputs_to_dict(self, input_example): |
| 403 | trt_inputs = {} |
| 404 | for i, inp in enumerate(input_example): |
| 405 | input_name = self.input_names[i] |
| 406 | trt_inputs[input_name] = inp |
| 407 | return trt_inputs |
| 408 | |
| 409 | def _load_engine(self): |
| 410 | """ |