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()