Searched refs:device_model (Results 1 – 1 of 1) sorted by relevance
11 torch::DeviceType device_model; variable57 PetscCall(EnumToDeviceType(device_enum, &device_model)); in LoadModel_Torch()60 PetscCallCXX(model.to(torch::Device(device_model))); in LoadModel_Torch()89 .to(device_model)); in ModelInference_Torch()