Lines Matching refs:comp
107 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in QPack2d() local
109 if (t_id_x < Q_1D && t_id_y < Q_1D) data.slice[t_id_x + t_id_y * T_1D] = U[comp]; in QPack2d()
111 U[comp] = data.t_id_x < (Q_1D * Q_1D) ? data.slice[new_t_id_x + new_t_id_y * T_1D] : 0.0; in QPack2d()
119 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in QUnpack2d() local
121 if (data.t_id_x < (Q_1D * Q_1D)) data.slice[old_t_id_x + old_t_id_y * T_1D] = U[comp]; in QUnpack2d()
123 U[comp] = (t_id_x < Q_1D && t_id_y < Q_1D) ? data.slice[t_id_x + t_id_y * T_1D] : 0.0; in QUnpack2d()
137 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTensor2dFlattened() local
138 ContractX2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp], c_B, r_t); in InterpTensor2dFlattened()
139 ContractY2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, r_t, c_B, &r_V[comp]); in InterpTensor2dFlattened()
156 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTransposeTensor2dFlattened() local
157 …ContractTransposeY2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp], c_B, r… in InterpTransposeTensor2dFlattened()
158 …tractTransposeX2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, r_t, c_B, &r_V[comp]); in InterpTransposeTensor2dFlattened()
173 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTensorCollocatedNodes2dFlattened() local
174 r_V[comp] = r_U[comp]; in InterpTensorCollocatedNodes2dFlattened()
190 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTransposeTensorCollocatedNodes2dFlattened() local
191 r_V[comp] = r_U[comp]; in InterpTransposeTensorCollocatedNodes2dFlattened()
207 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTensor2dFlattened() local
208 ContractX2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp], c_G, r_t); in GradTensor2dFlattened()
209 …ContractY2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, r_t, c_B, &r_V[comp + 0 * N… in GradTensor2dFlattened()
210 ContractX2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp], c_B, r_t); in GradTensor2dFlattened()
211 …ContractY2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, r_t, c_G, &r_V[comp + 1 * N… in GradTensor2dFlattened()
228 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTransposeTensor2dFlattened() local
229 …ContractTransposeY2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp + 0 * NU… in GradTransposeTensor2dFlattened()
230 …tractTransposeX2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, r_t, c_G, &r_V[comp]); in GradTransposeTensor2dFlattened()
231 …ContractTransposeY2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp + 1 * NU… in GradTransposeTensor2dFlattened()
232 …ctTransposeAddX2dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, r_t, c_B, &r_V[comp]); in GradTransposeTensor2dFlattened()
247 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTensorCollocatedNodes2dFlattened() local
248 …dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp], c_G, &r_V[comp + 0 * NUM_… in GradTensorCollocatedNodes2dFlattened()
249 …dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp], c_G, &r_V[comp + 1 * NUM_… in GradTensorCollocatedNodes2dFlattened()
265 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTransposeTensorCollocatedNodes2dFlattened() local
266 …ened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp + 1 * NUM_COMP], c_G, &r_V[comp]); in GradTransposeTensorCollocatedNodes2dFlattened()
267 …ened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, &r_U[comp + 0 * NUM_COMP], c_G, &r_V[comp]); in GradTransposeTensorCollocatedNodes2dFlattened()
444 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in QPack3d() local
446 …_id_x < Q_1D && t_id_y < Q_1D) data.slice[t_id_x + t_id_y * T_1D + t_id_z * T_1D * T_1D] = U[comp]; in QPack3d()
448 …U[comp] = data.t_id_x < (Q_1D * Q_1D * Q_1D) ? data.slice[new_t_id_x + new_t_id_y * T_1D + new_t_i… in QPack3d()
456 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in QUnpack3d() local
458 …_1D * Q_1D * Q_1D) data.slice[old_t_id_x + old_t_id_y * T_1D + old_t_id_z * T_1D * T_1D] = U[comp]; in QUnpack3d()
460 …U[comp] = (t_id_x < Q_1D && t_id_y < Q_1D) ? data.slice[t_id_x + t_id_y * T_1D + t_id_z * T_1D * T… in QUnpack3d()
474 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTensor3dFlattened() local
475 …ContractX3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp], c_B, r_… in InterpTensor3dFlattened()
477 …tractZ3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t2, c_B, &r_V[comp]); in InterpTensor3dFlattened()
494 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTransposeTensor3dFlattened() local
495 …sposeZ3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp], c_B, r_t1); in InterpTransposeTensor3dFlattened()
497 …sposeX3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t2, c_B, &r_V[comp]); in InterpTransposeTensor3dFlattened()
512 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTensorCollocatedNodes3dFlattened() local
513 r_V[comp] = r_U[comp]; in InterpTensorCollocatedNodes3dFlattened()
529 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in InterpTransposeTensorCollocatedNodes3dFlattened() local
530 r_V[comp] = r_U[comp]; in InterpTransposeTensorCollocatedNodes3dFlattened()
546 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTensor3dFlattened() local
547 …ContractX3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp], c_G, r_… in GradTensor3dFlattened()
549 …ed<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t2, c_B, &r_V[comp + 0 * NUM_COMP]); in GradTensor3dFlattened()
550 …ContractX3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp], c_B, r_… in GradTensor3dFlattened()
552 …ed<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t2, c_B, &r_V[comp + 1 * NUM_COMP]); in GradTensor3dFlattened()
553 …ContractX3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp], c_B, r_… in GradTensor3dFlattened()
555 …ed<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t2, c_G, &r_V[comp + 2 * NUM_COMP]); in GradTensor3dFlattened()
572 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTransposeTensor3dFlattened() local
573 …dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, &r_U[comp + 0 * NUM_COMP], c_… in GradTransposeTensor3dFlattened()
575 …sposeX3dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, r_t2, c_G, &r_V[comp]); in GradTransposeTensor3dFlattened()
576 …dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, &r_U[comp + 1 * NUM_COMP], c_… in GradTransposeTensor3dFlattened()
578 …seAddX3dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, r_t2, c_B, &r_V[comp]); in GradTransposeTensor3dFlattened()
579 …dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, &r_U[comp + 2 * NUM_COMP], c_… in GradTransposeTensor3dFlattened()
581 …seAddX3dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, r_t2, c_B, &r_V[comp]); in GradTransposeTensor3dFlattened()
597 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTensorCollocated3dFlattened() local
598 …ContractX3dFlattened<NUM_COMP, P_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp], c_B, r_… in GradTensorCollocated3dFlattened()
601 …ed<NUM_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t1, c_G, &r_V[comp + 0 * NUM_COMP]); in GradTensorCollocated3dFlattened()
602 …ed<NUM_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t1, c_G, &r_V[comp + 1 * NUM_COMP]); in GradTensorCollocated3dFlattened()
603 …ed<NUM_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_t1, c_G, &r_V[comp + 2 * NUM_COMP]); in GradTensorCollocated3dFlattened()
620 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTransposeTensor3dFlattened() local
621 …dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, &r_U[comp + 0 * NUM_COMP], c_… in GradTransposeTensor3dFlattened()
623 …sposeX3dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, r_t2, c_G, &r_V[comp]); in GradTransposeTensor3dFlattened()
624 …dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, &r_U[comp + 1 * NUM_COMP], c_… in GradTransposeTensor3dFlattened()
626 …seAddX3dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, r_t2, c_B, &r_V[comp]); in GradTransposeTensor3dFlattened()
627 …dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, &r_U[comp + 2 * NUM_COMP], c_… in GradTransposeTensor3dFlattened()
629 …seAddX3dFlattened<NUM_COMP, t_id_x, t_id_y, t_id_z, P_1D, Q_1D, T_1D>(data, r_t2, c_B, &r_V[comp]); in GradTransposeTensor3dFlattened()
644 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTensorCollocatedNodes3dFlattened() local
645 …ned<NUM_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_U[comp], c_G, &r_V[comp + 0 * NUM_… in GradTensorCollocatedNodes3dFlattened()
646 …ned<NUM_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_U[comp], c_G, &r_V[comp + 1 * NUM_… in GradTensorCollocatedNodes3dFlattened()
647 …ned<NUM_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, r_U[comp], c_G, &r_V[comp + 2 * NUM_… in GradTensorCollocatedNodes3dFlattened()
663 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradTransposeTensorCollocatedNodes3dFlattened() local
664 …_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp + 2 * NUM_COMP], c_G, &r_V[comp]); in GradTransposeTensorCollocatedNodes3dFlattened()
665 …_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp + 1 * NUM_COMP], c_G, &r_V[comp]); in GradTransposeTensorCollocatedNodes3dFlattened()
666 …_COMP, Q_1D, Q_1D, T_1D>(data, t_id_x, t_id_y, t_id_z, &r_U[comp + 0 * NUM_COMP], c_G, &r_V[comp]); in GradTransposeTensorCollocatedNodes3dFlattened()