Lines Matching refs:DD_Inputs_loc

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()
80     PetscCall(VecGetLocalSize(DD_Inputs_loc, &input_size));  in ModelInference_Torch()
82     PetscCall(VecGetArrayReadAndMemType(DD_Inputs_loc, &dd_inputs_ptr, &input_mem_type));  in ModelInference_Torch()
96       PetscCall(VecGetLocalSize(DD_Inputs_loc, &input_size));  in ModelInference_Torch()
103     PetscCall(VecRestoreArrayReadAndMemType(DD_Inputs_loc, &dd_inputs_ptr));  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()
155   PetscCall(PetscLogEventEnd(HONEE_SgsModelDDData, DD_Inputs_loc, DD_Outputs_loc, NULL, NULL));  in ModelInference_Torch()