Home
last modified time | relevance | path

Searched refs:stride_1 (Results 1 – 3 of 3) sorted by relevance

/libCEED/include/ceed/jit-source/hip/
H A Dhip-ref-basis-tensor.h34 const CeedInt stride_1 = is_transpose ? BASIS_P_1D : 1; in Interp() local
64 …for (CeedInt b = 0; b < P; b++) v_k += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) *… in Interp()
94 const CeedInt stride_1 = is_transpose ? BASIS_P_1D : 1; in Grad() local
128 …for (CeedInt b = 0; b < P; b++) v_k += op[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c… in Grad()
/libCEED/include/ceed/jit-source/cuda/
H A Dcuda-ref-basis-tensor.h34 const CeedInt stride_1 = is_transpose ? BASIS_P_1D : 1; in Interp() local
64 …for (CeedInt b = 0; b < P; b++) v_k += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) *… in Interp()
94 const CeedInt stride_1 = is_transpose ? BASIS_P_1D : 1; in Grad() local
128 …for (CeedInt b = 0; b < P; b++) v_k += op[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c… in Grad()
/libCEED/backends/sycl-ref/
H A Dceed-sycl-ref-basis.sycl.cpp75 const CeedInt stride_1 = is_transpose ? P_1d : 1; in CeedBasisApplyInterp_Sycl() local
126 vk += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c]; in CeedBasisApplyInterp_Sycl()
178 const CeedInt stride_1 = is_transpose ? P_1d : 1; in CeedBasisApplyGrad_Sycl() local
227 …for (CeedInt b = 0; b < P; b++) v_k += op[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c… in CeedBasisApplyGrad_Sycl()
344 const CeedInt stride_1 = is_transpose ? impl->num_nodes : 1; in CeedBasisApplyNonTensorInterp_Sycl() local
368 V += d_B[i * stride_0 + j * stride_1] * U[j]; in CeedBasisApplyNonTensorInterp_Sycl()
385 const CeedInt stride_1 = is_transpose ? impl->num_nodes : 1; in CeedBasisApplyNonTensorGrad_Sycl() local
420 V[d0] += G[j * stride_1 + d0 * g_dim_stride] * Uj; in CeedBasisApplyNonTensorGrad_Sycl()