Lines Matching refs:t_id_y
131 data.slice[data.t_id_x + data.t_id_y * T_1D] = *U; in ContractX2d()
134 if (data.t_id_x < Q_1D && data.t_id_y < P_1D) { in ContractX2d()
136 *V += B[i + data.t_id_x * P_1D] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractX2d()
147 data.slice[data.t_id_x + data.t_id_y * T_1D] = *U; in ContractY2d()
150 if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) { in ContractY2d()
152 *V += B[i + data.t_id_y * P_1D] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractY2d()
163 data.slice[data.t_id_x + data.t_id_y * T_1D] = *U; in ContractTransposeY2d()
166 if (data.t_id_x < Q_1D && data.t_id_y < P_1D) { in ContractTransposeY2d()
168 *V += B[data.t_id_y + i * P_1D] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractTransposeY2d()
179 data.slice[data.t_id_x + data.t_id_y * T_1D] = *U; in ContractTransposeX2d()
182 if (data.t_id_x < P_1D && data.t_id_y < P_1D) { in ContractTransposeX2d()
184 *V += B[data.t_id_x + i * P_1D] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractTransposeX2d()
195 data.slice[data.t_id_x + data.t_id_y * T_1D] = *U; in ContractTransposeAddX2d()
197 if (data.t_id_x < P_1D && data.t_id_y < P_1D) { in ContractTransposeAddX2d()
199 *V += B[data.t_id_x + i * P_1D] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractTransposeAddX2d()
311 …*w = (data.t_id_x < Q_1D && data.t_id_y < Q_1D) ? q_weight_1d[data.t_id_x] * q_weight_1d[data.t_id… in WeightTensor2d()
330 data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k]; in ContractX3d()
333 if (data.t_id_x < Q_1D && data.t_id_y < P_1D) { in ContractX3d()
335 V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractX3d()
348 r_B[i] = B[i + data.t_id_y * P_1D]; in ContractY3d()
353 data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k]; in ContractY3d()
356 if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) { in ContractY3d()
371 if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) { in ContractZ3d()
386 if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) { in ContractTransposeZ3d()
401 r_B[i] = B[data.t_id_y + i * P_1D]; in ContractTransposeY3d()
406 data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k]; in ContractTransposeY3d()
409 if (data.t_id_x < Q_1D && data.t_id_y < P_1D) { in ContractTransposeY3d()
424 r_B[i] = B[data.t_id_y + i * P_1D]; in ContractTransposeAddY3d()
429 data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k]; in ContractTransposeAddY3d()
431 if (data.t_id_x < Q_1D && data.t_id_y < P_1D) { in ContractTransposeAddY3d()
451 data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k]; in ContractTransposeX3d()
454 if (data.t_id_x < P_1D && data.t_id_y < P_1D) { in ContractTransposeX3d()
456 V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractTransposeX3d()
474 data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k]; in ContractTransposeAddX3d()
476 if (data.t_id_x < P_1D && data.t_id_y < P_1D) { in ContractTransposeAddX3d()
478 V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractTransposeAddX3d()
649 const bool quad = (data.t_id_x < Q_1D && data.t_id_y < Q_1D); in WeightTensor3d()
650 const CeedScalar pw = quad ? q_weight_1d[data.t_id_x] * q_weight_1d[data.t_id_y] : 0.0; in WeightTensor3d()