Lines Matching refs:B
20 inline __device__ void ContractX1d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, … in ContractX1d() argument
27 *V += B[i + data.t_id_x * P_1D] * data.slice[i]; // Contract x direction in ContractX1d()
36 …ntractTransposeX1d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeX1d() argument
43 *V += B[data.t_id_x + i * P_1D] * data.slice[i]; // Contract x direction in ContractTransposeX1d()
129 inline __device__ void ContractX2d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, … in ContractX2d() argument
136 *V += B[i + data.t_id_x * P_1D] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractX2d()
145 inline __device__ void ContractY2d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, … in ContractY2d() argument
152 *V += B[i + data.t_id_y * P_1D] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractY2d()
161 …ntractTransposeY2d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeY2d() argument
168 *V += B[data.t_id_y + i * P_1D] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractTransposeY2d()
177 …ntractTransposeX2d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeX2d() argument
184 *V += B[data.t_id_x + i * P_1D] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractTransposeX2d()
193 …actTransposeAddX2d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeAddX2d() argument
199 *V += B[data.t_id_x + i * P_1D] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractTransposeAddX2d()
321 inline __device__ void ContractX3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, … in ContractX3d() argument
324 r_B[i] = B[i + data.t_id_x * P_1D]; in ContractX3d()
344 inline __device__ void ContractY3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, … in ContractY3d() argument
347 r_B[i] = B[i + data.t_id_y * P_1D]; in ContractY3d()
367 inline __device__ void ContractZ3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, … in ContractZ3d() argument
372 V[k] += B[i + k * P_1D] * U[i]; // Contract z direction in ContractZ3d()
382 …ntractTransposeZ3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeZ3d() argument
387 V[k] += B[k + i * P_1D] * U[i]; // Contract z direction in ContractTransposeZ3d()
397 …ntractTransposeY3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeY3d() argument
400 r_B[i] = B[data.t_id_y + i * P_1D]; in ContractTransposeY3d()
420 …actTransposeAddY3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeAddY3d() argument
423 r_B[i] = B[data.t_id_y + i * P_1D]; in ContractTransposeAddY3d()
442 …ntractTransposeX3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeX3d() argument
445 r_B[i] = B[data.t_id_x + i * P_1D]; in ContractTransposeX3d()
465 …actTransposeAddX3d(SharedData_Hip &data, const CeedScalar *U, const CeedScalar *B, CeedScalar *V) { in ContractTransposeAddX3d() argument
468 r_B[i] = B[data.t_id_x + i * P_1D]; in ContractTransposeAddX3d()