Patch function to replace original model's forward() with. Redirects to TrtCompiler.forward()
(self, *argv, **kwargs)
| 602 | |
| 603 | |
| 604 | def trt_forward(self, *argv, **kwargs): |
| 605 | """ |
| 606 | Patch function to replace original model's forward() with. |
| 607 | Redirects to TrtCompiler.forward() |
| 608 | """ |
| 609 | return self._trt_compiler.forward(self, argv, kwargs) |
| 610 | |
| 611 | |
| 612 | def trt_compile( |
nothing calls this directly
no test coverage detected
searching dependent graphs…