Searched refs:PetscMemTypeToDeviceType (Results 1 – 1 of 1) sorted by relevance
34 static PetscErrorCode PetscMemTypeToDeviceType(PetscMemType mem_type, torch::DeviceType *device_typ… in PetscMemTypeToDeviceType() function83 PetscCall(PetscMemTypeToDeviceType(input_mem_type, &dd_input_device)); in ModelInference_Torch()124 PetscCall(PetscMemTypeToDeviceType(output_mem_type, &dd_output_device)); in ModelInference_Torch()