Searched refs:LoadModel_Torch (Results 1 – 4 of 4) sorted by relevance
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() function
268 PetscCall(LoadModel_Torch(model_path, model_device_type)); in SgsDDSetupNodalEvaluation_Sequential_Torch()
18 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum);
55 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) { in LoadModel_Torch() function