Searched refs:EnumToDeviceType (Results 1 – 1 of 1) sorted by relevance
13 static PetscErrorCode EnumToDeviceType(TorchDeviceType device_enum, torch::DeviceType *device_type)… in EnumToDeviceType() function57 PetscCall(EnumToDeviceType(device_enum, &device_model)); in LoadModel_Torch()