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()
310 …*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()
329     data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k];  in ContractX3d()
332     if (data.t_id_x < Q_1D && data.t_id_y < P_1D) {  in ContractX3d()
334         V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D];  // Contract x direction  in ContractX3d()
347     r_B[i] = B[i + data.t_id_y * P_1D];  in ContractY3d()
352     data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k];  in ContractY3d()
355     if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) {  in ContractY3d()
370     if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) {  in ContractZ3d()
385     if (data.t_id_x < Q_1D && data.t_id_y < Q_1D) {  in ContractTransposeZ3d()
400     r_B[i] = B[data.t_id_y + i * P_1D];  in ContractTransposeY3d()
405     data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k];  in ContractTransposeY3d()
408     if (data.t_id_x < Q_1D && data.t_id_y < P_1D) {  in ContractTransposeY3d()
423     r_B[i] = B[data.t_id_y + i * P_1D];  in ContractTransposeAddY3d()
428     data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k];  in ContractTransposeAddY3d()
430     if (data.t_id_x < Q_1D && data.t_id_y < P_1D) {  in ContractTransposeAddY3d()
450     data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k];  in ContractTransposeX3d()
453     if (data.t_id_x < P_1D && data.t_id_y < P_1D) {  in ContractTransposeX3d()
455         V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D];  // Contract x direction  in ContractTransposeX3d()
473     data.slice[data.t_id_x + data.t_id_y * T_1D] = U[k];  in ContractTransposeAddX3d()
475     if (data.t_id_x < P_1D && data.t_id_y < P_1D) {  in ContractTransposeAddX3d()
477         V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D];  // Contract x direction  in ContractTransposeAddX3d()
647   const bool       quad = (data.t_id_x < Q_1D && data.t_id_y < Q_1D);  in WeightTensor3d()
648   const CeedScalar pw   = quad ? q_weight_1d[data.t_id_x] * q_weight_1d[data.t_id_y] : 0.0;  in WeightTensor3d()