Searched defs:device_enum (Results 1 – 2 of 2) sorted by relevance
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()
9 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) { in LoadModel_Torch()