Searched refs:input_mem_type (Results 1 – 1 of 1) sorted by relevance
74 PetscMemType input_mem_type; in ModelInference_Torch() local82 PetscCall(VecGetArrayReadAndMemType(DD_Inputs_loc, &dd_inputs_ptr, &input_mem_type)); in ModelInference_Torch()83 PetscCall(PetscMemTypeToDeviceType(input_mem_type, &dd_input_device)); in ModelInference_Torch()