(self, gpu_memory_mb)
| 294 | self.trt_dynamic_shape_info = info |
| 295 | |
| 296 | def set_gpu_memory_mb(self, gpu_memory_mb): |
| 297 | self.gpu_memory_mb = gpu_memory_mb |
| 298 | |
| 299 | def set_cpu_math_thread_num(self, cpu_math_thread_num): |
| 300 | self.cpu_math_thread_num = cpu_math_thread_num |
no outgoing calls
no test coverage detected