Lines Matching refs:i
34 for (CeedInt i = 0; i < N; i++) x[i] *= (x[i] < 0 ? alpha : 1.); in LeakyReLU() local
78 CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { in ComputeSgsDDNodal_Fused() local
79 const CeedScalar qi[5] = {q[0][i], q[1][i], q[2][i], q[3][i], q[4][i]}; in ComputeSgsDDNodal_Fused()
81 {grad_velo[0][0][i], grad_velo[0][1][i], grad_velo[0][2][i]}, in ComputeSgsDDNodal_Fused()
82 {grad_velo[1][0][i], grad_velo[1][1][i], grad_velo[1][2][i]}, in ComputeSgsDDNodal_Fused()
83 {grad_velo[2][0][i], grad_velo[2][1][i], grad_velo[2][2][i]} in ComputeSgsDDNodal_Fused()
85 …ar km_A_ij[6] = {A_ij_delta[0][i], A_ij_delta[1][i], A_ij_delta[2][i], A_ij_delta[3][i], A_ij_delt… in ComputeSgsDDNodal_Fused()
86 const CeedScalar delta = A_ij_delta[6][i]; in ComputeSgsDDNodal_Fused()
92 for (int j = 0; j < 6; j++) v[j][i] = inv_multiplicity[i] * km_sgs[j]; in ComputeSgsDDNodal_Fused()
122 CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { in ComputeSgsDDNodal_Sequential_Inputs() local
123 const CeedScalar qi[5] = {q[0][i], q[1][i], q[2][i], q[3][i], q[4][i]}; in ComputeSgsDDNodal_Sequential_Inputs()
125 {grad_velo[0][0][i], grad_velo[0][1][i], grad_velo[0][2][i]}, in ComputeSgsDDNodal_Sequential_Inputs()
126 {grad_velo[1][0][i], grad_velo[1][1][i], grad_velo[1][2][i]}, in ComputeSgsDDNodal_Sequential_Inputs()
127 {grad_velo[2][0][i], grad_velo[2][1][i], grad_velo[2][2][i]} in ComputeSgsDDNodal_Sequential_Inputs()
129 …ar km_A_ij[6] = {A_ij_delta[0][i], A_ij_delta[1][i], A_ij_delta[2][i], A_ij_delta[3][i], A_ij_delt… in ComputeSgsDDNodal_Sequential_Inputs()
130 const CeedScalar delta = A_ij_delta[6][i]; in ComputeSgsDDNodal_Sequential_Inputs()
136 ScaleN(model_inputs_i, inv_multiplicity[i], 6); in ComputeSgsDDNodal_Sequential_Inputs()
137 StoredValuesPack(Q, i, 0, 6, model_inputs_i, (CeedScalar *)model_inputs); in ComputeSgsDDNodal_Sequential_Inputs()
138 StoredValuesPack(Q, i, 0, 9, (const CeedScalar *)eigenvectors, eigenvectors_stored); in ComputeSgsDDNodal_Sequential_Inputs()
139 StoredValuesPack(Q, i, 9, 1, &grad_velo_magnitude, eigenvectors_stored); in ComputeSgsDDNodal_Sequential_Inputs()
164 CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { in ComputeSgsDDNodal_Sequential_Inference() local
167 StoredValuesUnpack(Q, i, 0, 6, (const CeedScalar *)model_inputs, model_inputs_i); in ComputeSgsDDNodal_Sequential_Inference()
169 ScaleN(model_outputs_i, inv_multiplicity[i], 6); in ComputeSgsDDNodal_Sequential_Inference()
170 StoredValuesPack(Q, i, 0, 6, model_outputs_i, model_outputs); in ComputeSgsDDNodal_Sequential_Inference()
187 CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { in ComputeSgsDDNodal_Sequential_Outputs() local
189 const CeedScalar delta = A_ij_delta[6][i]; in ComputeSgsDDNodal_Sequential_Outputs()
191 StoredValuesUnpack(Q, i, 0, 6, model_outputs, model_outputs_i); in ComputeSgsDDNodal_Sequential_Outputs()
193 StoredValuesUnpack(Q, i, 0, 9, eigenvectors_stored, (CeedScalar *)eigenvectors); in ComputeSgsDDNodal_Sequential_Outputs()
194 StoredValuesUnpack(Q, i, 9, 1, eigenvectors_stored, &grad_velo_magnitude); in ComputeSgsDDNodal_Sequential_Outputs()
197 for (int j = 0; j < 6; j++) kmsgs_stress[j][i] = inv_multiplicity[i] * kmsgs_stress_i[j]; in ComputeSgsDDNodal_Sequential_Outputs()
223 CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { in IFunction_NodalSgs() local
224 const CeedScalar qi[5] = {q[0][i], q[1][i], q[2][i], q[3][i], q[4][i]}; in IFunction_NodalSgs()
228 QdataUnpack_3D(Q, i, q_data, &wdetJ, dXdx); in IFunction_NodalSgs()
231 …onst CeedScalar km_sgs_i[6] = {km_sgs[0][i], km_sgs[1][i], km_sgs[2][i], km_sgs[3][i], km_sgs[4][i… in IFunction_NodalSgs()
236 …Grad_v[k][j][i] = -wdetJ * (dXdx[k][0] * Flux[j][0] + dXdx[k][1] * Flux[j][1] + dXdx[k][2] * Flux[… in IFunction_NodalSgs()