(self, reuse_device_memory=None)
| 222 | self.engine = engine_from_bytes(bytes_from_path(self.engine_path)) |
| 223 | |
| 224 | def activate(self, reuse_device_memory=None): |
| 225 | if reuse_device_memory: |
| 226 | self.context = self.engine.create_execution_context_without_device_memory() |
| 227 | self.context.device_memory = reuse_device_memory |
| 228 | else: |
| 229 | self.context = self.engine.create_execution_context() |
| 230 | |
| 231 | def allocate_buffers(self, shape_dict=None, device='cuda'): |
| 232 | for idx in range(trt_util.get_bindings_per_profile(self.engine)): |
no outgoing calls
no test coverage detected