Searched refs:output_size (Results 1 – 2 of 2) sorted by relevance
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 118 PetscInt output_size; in ModelInference_Torch() local
131 PetscCall(VecGetLocalSize(DD_Outputs_loc, &output_size)); in ModelInference_Torch()
136 for (PetscInt i = 0; i < output_size; i++) { in ModelInference_Torch()
140 PetscCall(PetscArraycpy(dd_outputs_ptr, output_tensor_ptr, output_size)); in ModelInference_Torch()
146 PetscCall(VecGetLocalSize(DD_Outputs_loc, &output_size)); in ModelInference_Torch()
147 num_nodes = output_size / num_output_comps; in ModelInference_Torch()
|
| /honee/src/ |
| H A D | petsc_ops.c | 173 CeedSize input_size, output_size; in CeedOperatorCreateLocalVecs() local
181 PetscCallCeed(ceed, CeedOperatorGetActiveVectorLengths(op, &input_size, &output_size)); in CeedOperatorCreateLocalVecs()
190 PetscCall(VecSetSizes(*output, output_size, output_size)); in CeedOperatorCreateLocalVecs()
|