Searched refs:dd_inputs_ptr (Results 1 – 1 of 1) sorted by relevance
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 76 const PetscScalar *dd_inputs_ptr; in ModelInference_Torch() local
82 PetscCall(VecGetArrayReadAndMemType(DD_Inputs_loc, &dd_inputs_ptr, &input_mem_type)); in ModelInference_Torch()
88 …at::from_blob((void *)dd_inputs_ptr, {num_nodes, num_input_comps}, {num_input_comps, 1}, nullptr, … in ModelInference_Torch()
91 …PetscCallCXX(input_tensor = torch::from_blob((void *)dd_inputs_ptr, {num_nodes, num_input_comps}, … in ModelInference_Torch()
103 PetscCall(VecRestoreArrayReadAndMemType(DD_Inputs_loc, &dd_inputs_ptr)); in ModelInference_Torch()
|