Searched refs:output_mem_type (Results 1 – 1 of 1) sorted by relevance
120 PetscMemType output_mem_type; in ModelInference_Torch() local123 PetscCall(VecGetArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr, &output_mem_type)); in ModelInference_Torch()124 PetscCall(PetscMemTypeToDeviceType(output_mem_type, &dd_output_device)); in ModelInference_Torch()148 PetscCall(VecGetArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr, &output_mem_type)); in ModelInference_Torch()