Searched refs:input_size (Results 1 – 2 of 2) sorted by relevance
75 PetscInt input_size, num_nodes; in ModelInference_Torch() local80 PetscCall(VecGetLocalSize(DD_Inputs_loc, &input_size)); in ModelInference_Torch()81 num_nodes = input_size / num_input_comps; in ModelInference_Torch()96 PetscCall(VecGetLocalSize(DD_Inputs_loc, &input_size)); in ModelInference_Torch()99 for (PetscInt i = 0; i < input_size; i++) { in ModelInference_Torch()
173 CeedSize input_size, output_size; in CeedOperatorCreateLocalVecs() local181 PetscCallCeed(ceed, CeedOperatorGetActiveVectorLengths(op, &input_size, &output_size)); in CeedOperatorCreateLocalVecs()185 PetscCall(VecSetSizes(*input, input_size, input_size)); in CeedOperatorCreateLocalVecs()