Searched refs:dd_outputs_ptr (Results 1 – 1 of 1) sorted by relevance
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 119 PetscScalar *dd_outputs_ptr; in ModelInference_Torch() local
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()
132 PetscCall(VecGetArray(DD_Outputs_loc, &dd_outputs_ptr)); in ModelInference_Torch()
140 PetscCall(PetscArraycpy(dd_outputs_ptr, output_tensor_ptr, output_size)); in ModelInference_Torch()
141 PetscCall(VecRestoreArray(DD_Outputs_loc, &dd_outputs_ptr)); in ModelInference_Torch()
148 PetscCall(VecGetArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr, &output_mem_type)); in ModelInference_Torch()
150 …PetscCallCXX(DD_Outputs_tensor = torch::from_blob((void *)dd_outputs_ptr, {num_nodes, num_output_c… in ModelInference_Torch()
152 PetscCall(VecRestoreArrayAndMemType(DD_Outputs_loc, &dd_outputs_ptr)); in ModelInference_Torch()
|