Lines Matching refs:item_id_x
21 const CeedInt item_id_x = get_local_id(0); in ContractX1d() local
23 scratch[item_id_x] = *U; in ContractX1d()
27 if (item_id_x < Q_1D) { in ContractX1d()
29 *V += B[i + item_id_x * P_1D] * scratch[i]; // Contract x direction in ContractX1d()
40 const CeedInt item_id_x = get_local_id(0); in ContractTransposeX1d() local
42 scratch[item_id_x] = *U; in ContractTransposeX1d()
46 if (item_id_x < P_1D) { in ContractTransposeX1d()
48 *V += B[item_id_x + i * P_1D] * scratch[i]; // Contract x direction in ContractTransposeX1d()
98 const CeedInt item_id_x = get_local_id(0); in Weight1d() local
99 *w = (item_id_x < Q_1D) ? q_weight_1d[item_id_x] : 0.0; in Weight1d()
111 const CeedInt item_id_x = get_local_id(0); 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()
131 const CeedInt item_id_x = get_local_id(0); 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()
151 const CeedInt item_id_x = get_local_id(0); 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()
171 const CeedInt item_id_x = get_local_id(0); 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()
191 const CeedInt item_id_x = get_local_id(0); 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()
267 const CeedInt item_id_x = get_local_id(0); 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()
282 const CeedInt item_id_x = get_local_id(0); in ContractX3d() local
287 r_B[i] = B[i + item_id_x * P_1D]; in ContractX3d()
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()
309 const CeedInt item_id_x = get_local_id(0); in ContractY3d() local
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()
324 V[k] += r_B[i] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractY3d()
336 const CeedInt item_id_x = get_local_id(0); in ContractZ3d() local
341 if (item_id_x < Q_1D && item_id_y < Q_1D) { in ContractZ3d()
354 const CeedInt item_id_x = get_local_id(0); in ContractTransposeZ3d() local
359 if (item_id_x < Q_1D && item_id_y < Q_1D) { in ContractTransposeZ3d()
372 const CeedInt item_id_x = get_local_id(0); in ContractTransposeY3d() local
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()
387 V[k] += r_B[i] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractTransposeY3d()
399 const CeedInt item_id_x = get_local_id(0); in ContractTransposeAddY3d() local
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()
412 V[k] += r_B[i] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractTransposeAddY3d()
424 const CeedInt item_id_x = get_local_id(0); in ContractTransposeX3d() local
429 r_B[i] = B[item_id_x + i * P_1D]; in ContractTransposeX3d()
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()
450 const CeedInt item_id_x = get_local_id(0); in ContractTransposeAddX3d() local
455 r_B[i] = B[item_id_x + i * P_1D]; in ContractTransposeAddX3d()
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()
588 const CeedInt item_id_x = get_local_id(0); 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()