Searched refs:TorchDeviceType (Results 1 – 4 of 4) sorted by relevance
15 } TorchDeviceType; typedef18 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum);
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()
256 TorchDeviceType model_device_type; in SgsDDSetupNodalEvaluation_Sequential_Torch()
13 static PetscErrorCode EnumToDeviceType(TorchDeviceType device_enum, torch::DeviceType *device_type)… in EnumToDeviceType()55 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) { in LoadModel_Torch()