Lines Matching refs:item_id_y
112 const CeedInt item_id_y = get_local_id(1); in ContractX2d() local
114 scratch[item_id_x + item_id_y * T_1D] = *U; in ContractX2d()
118 if (item_id_x < Q_1D && item_id_y < P_1D) { in ContractX2d()
120 *V += B[i + item_id_x * P_1D] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractX2d()
132 const CeedInt item_id_y = get_local_id(1); in ContractY2d() local
134 scratch[item_id_x + item_id_y * T_1D] = *U; in ContractY2d()
138 if (item_id_x < Q_1D && item_id_y < Q_1D) { in ContractY2d()
140 *V += B[i + item_id_y * P_1D] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractY2d()
152 const CeedInt item_id_y = get_local_id(1); in ContractTransposeY2d() local
154 scratch[item_id_x + item_id_y * T_1D] = *U; in ContractTransposeY2d()
158 if (item_id_x < Q_1D && item_id_y < P_1D) { in ContractTransposeY2d()
160 *V += B[item_id_y + i * P_1D] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractTransposeY2d()
172 const CeedInt item_id_y = get_local_id(1); in ContractTransposeX2d() local
174 scratch[item_id_x + item_id_y * T_1D] = *U; in ContractTransposeX2d()
178 if (item_id_x < P_1D && item_id_y < P_1D) { in ContractTransposeX2d()
180 *V += B[item_id_x + i * P_1D] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractTransposeX2d()
192 const CeedInt item_id_y = get_local_id(1); in ContractTransposeAddX2d() local
194 scratch[item_id_x + item_id_y * T_1D] = *U; in ContractTransposeAddX2d()
197 if (item_id_x < P_1D && item_id_y < P_1D) { in ContractTransposeAddX2d()
199 *V += B[item_id_x + i * P_1D] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractTransposeAddX2d()
268 const CeedInt item_id_y = get_local_id(1); in WeightTensor2d() local
270 …*w = (item_id_x < Q_1D && item_id_y < Q_1D) ? q_weight_1d[item_id_x] * q_weight_1d[item_id_y] : 0.… in WeightTensor2d()
283 const CeedInt item_id_y = get_local_id(1); in ContractX3d() local
291 scratch[item_id_x + item_id_y * T_1D] = U[k]; in ContractX3d()
295 if (item_id_x < Q_1D && item_id_y < P_1D) { in ContractX3d()
297 V[k] += r_B[i] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractX3d()
310 const CeedInt item_id_y = get_local_id(1); in ContractY3d() local
314 r_B[i] = B[i + item_id_y * P_1D]; in ContractY3d()
318 scratch[item_id_x + item_id_y * T_1D] = U[k]; in ContractY3d()
322 if (item_id_x < Q_1D && item_id_y < Q_1D) { in ContractY3d()
337 const CeedInt item_id_y = get_local_id(1); in ContractZ3d() local
341 if (item_id_x < Q_1D && item_id_y < Q_1D) { in ContractZ3d()
355 const CeedInt item_id_y = get_local_id(1); in ContractTransposeZ3d() local
359 if (item_id_x < Q_1D && item_id_y < Q_1D) { in ContractTransposeZ3d()
373 const CeedInt item_id_y = get_local_id(1); in ContractTransposeY3d() local
377 r_B[i] = B[item_id_y + i * P_1D]; in ContractTransposeY3d()
381 scratch[item_id_x + item_id_y * T_1D] = U[k]; in ContractTransposeY3d()
385 if (item_id_x < Q_1D && item_id_y < P_1D) { in ContractTransposeY3d()
400 const CeedInt item_id_y = get_local_id(1); in ContractTransposeAddY3d() local
404 r_B[i] = B[item_id_y + i * P_1D]; in ContractTransposeAddY3d()
408 scratch[item_id_x + item_id_y * T_1D] = U[k]; in ContractTransposeAddY3d()
410 if (item_id_x < Q_1D && item_id_y < P_1D) { in ContractTransposeAddY3d()
425 const CeedInt item_id_y = get_local_id(1); in ContractTransposeX3d() local
433 scratch[item_id_x + item_id_y * T_1D] = U[k]; in ContractTransposeX3d()
436 if (item_id_x < P_1D && item_id_y < P_1D) { in ContractTransposeX3d()
438 V[k] += r_B[i] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractTransposeX3d()
451 const CeedInt item_id_y = get_local_id(1); in ContractTransposeAddX3d() local
459 scratch[item_id_x + item_id_y * T_1D] = U[k]; in ContractTransposeAddX3d()
462 if (item_id_x < P_1D && item_id_y < P_1D) { in ContractTransposeAddX3d()
464 V[k] += r_B[i] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractTransposeAddX3d()
589 const CeedInt item_id_y = get_local_id(1); in WeightTensor3d() local
591 if (item_id_x < Q_1D && item_id_y < Q_1D) { in WeightTensor3d()
592 const CeedScalar w_xy = q_weight_1d[item_id_x] * q_weight_1d[item_id_y]; in WeightTensor3d()