Searched refs:model_path (Results 1 – 4 of 4) sorted by relevance
| /honee/problems/ |
| H A D | sgs_model_torch_weak.c | 8 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) __attribute__((…
9 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) { in LoadModel_Torch() argument
|
| H A D | sgs_dd_model.c | 255 char model_path[PETSC_MAX_PATH_LEN] = ""; in SgsDDSetupNodalEvaluation_Sequential_Torch() local
266 …ptionsGetString(NULL, NULL, "-sgs_model_dd_torch_model_path", model_path, sizeof(model_path), NULL… in SgsDDSetupNodalEvaluation_Sequential_Torch()
268 PetscCall(LoadModel_Torch(model_path, model_device_type)); in SgsDDSetupNodalEvaluation_Sequential_Torch()
|
| /honee/include/ |
| H A D | sgs_model_torch.h | 18 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum);
|
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 55 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) { in LoadModel_Torch() argument
59 PetscCallCXX(model = torch::jit::load(model_path)); in LoadModel_Torch()
|