Lines Matching refs:P
58 template <typename T, int P, int DIM_U, int NUM_COMP, int rU_SIZE, int i_DIM>
72 if (tx < P) { in read_U_2d()
73 for (int i = 0; i < P; i++) { in read_U_2d()
74 sTmp[i * P + tx] = dU[comp * compstride + i * P + tx]; in read_U_2d()
79 if (tx < P) { in read_U_2d()
80 for (int i = 0; i < P; i++) { in read_U_2d()
81 rU[i_DIM][comp][i] = sTmp[tx * P + i]; in read_U_2d()
146 template <typename T, int P, int DIM_U, int NUM_COMP, int rU_SIZE, int i_DIM>
160 if (tx < P * P) { in read_U_3d()
161 for (int i = 0; i < P; i++) { in read_U_3d()
162 sTmp[i * P * P + tx] = dU[comp * compstride + i * P * P + tx]; in read_U_3d()
167 if (tx < P * P) { in read_U_3d()
168 for (int i = 0; i < P; i++) { in read_U_3d()
169 rU[i_DIM][comp][i] = sTmp[tx * P + i]; in read_U_3d()