Searched refs:num_nodes (Results 1 – 2 of 2) sorted by relevance
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 75 PetscInt input_size, num_nodes; in ModelInference_Torch() local
81 num_nodes = input_size / num_input_comps; 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()
143 PetscInt num_nodes; in ModelInference_Torch() local
147 num_nodes = output_size / num_output_comps; in ModelInference_Torch()
150 …PetscCallCXX(DD_Outputs_tensor = torch::from_blob((void *)dd_outputs_ptr, {num_nodes, num_output_c… in ModelInference_Torch()
|
| /honee/src/ |
| H A D | grid_anisotropy_tensor.c | 127 CeedInt q_data_size, num_nodes; in GridAnisotropyTensorCalculateCollocatedVector() local
139 PetscCallCeed(ceed, CeedBasisGetNumNodes(basis_q, &num_nodes)); in GridAnisotropyTensorCalculateCollocatedVector()
|