(model_name)
| 44 | |
| 45 | @staticmethod |
| 46 | def get_onnx_path(model_name): |
| 47 | onnx_filename = f"{model_name}.onnx" |
| 48 | onnx_path = os.path.join(ONNX_MODEL_DIR, onnx_filename) |
| 49 | return onnx_filename, onnx_path |
| 50 | |
| 51 | def get_trt_path(self, model_name, model_hash, profile, static_shape): |
| 52 | profile_hash = [] |
no outgoing calls
no test coverage detected