Searched refs:device_enum (Results 1 – 3 of 3) sorted by relevance
13 static PetscErrorCode EnumToDeviceType(TorchDeviceType device_enum, torch::DeviceType *device_type)… in EnumToDeviceType() argument15 switch (device_enum) { in EnumToDeviceType()29 …C_COMM_WORLD, PETSC_ERR_SUP, "TorchDeviceType %d not supported by PyTorch inference", device_enum); in EnumToDeviceType()55 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum) { in LoadModel_Torch() argument57 PetscCall(EnumToDeviceType(device_enum, &device_model)); in LoadModel_Torch()
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
18 PetscErrorCode LoadModel_Torch(const char *model_path, TorchDeviceType device_enum);