65 PetscErrorCode ModelInference_Torch(Vec DD_Inputs_loc, Vec DD_Outputs_loc) {  in ModelInference_Torch()  argument
72   PetscCall(PetscLogEventBegin(HONEE_SgsModelDDData, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL));  in ModelInference_Torch()
105   PetscCall(PetscLogEventEnd(HONEE_SgsModelDDData, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL));  in ModelInference_Torch()
108 …PetscCall(PetscLogEventBegin(HONEE_SgsModelDDInference, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL)…  in ModelInference_Torch()
112   PetscCall(PetscLogEventEnd(HONEE_SgsModelDDInference, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL));  in ModelInference_Torch()
114   PetscCall(PetscLogEventBegin(HONEE_SgsModelDDData, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL));  in ModelInference_Torch()
123       PetscCall(VecGetArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr, &output_mem_type));  in ModelInference_Torch()
125       PetscCall(VecRestoreArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr));  in ModelInference_Torch()
131       PetscCall(VecGetLocalSize(DD_Outputs_loc, &output_size));  in ModelInference_Torch()
132       PetscCall(VecGetArray(DD_Outputs_loc, &dd_outputs_ptr));  in ModelInference_Torch()
141       PetscCall(VecRestoreArray(DD_Outputs_loc, &dd_outputs_ptr));  in ModelInference_Torch()
146       PetscCall(VecGetLocalSize(DD_Outputs_loc, &output_size));  in ModelInference_Torch()
148       PetscCall(VecGetArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr, &output_mem_type));  in ModelInference_Torch()
152       PetscCall(VecRestoreArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr));  in ModelInference_Torch()
155   PetscCall(PetscLogEventEnd(HONEE_SgsModelDDData, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL));  in ModelInference_Torch()