Home
last modified time | relevance | path

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

/libCEED/include/ceed/jit-source/hip/
H A Dhip-ref-basis-tensor.h33 const CeedInt stride_0 = is_transpose ? 1 : BASIS_P_1D; 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()
93 const CeedInt stride_0 = is_transpose ? 1 : BASIS_P_1D; 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.h33 const CeedInt stride_0 = is_transpose ? 1 : BASIS_P_1D; 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()
93 const CeedInt stride_0 = is_transpose ? 1 : BASIS_P_1D; 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.cpp74 const CeedInt stride_0 = is_transpose ? 1 : P_1d; in CeedBasisApplyInterp_Sycl() local
126 vk += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c]; in CeedBasisApplyInterp_Sycl()
177 const CeedInt stride_0 = is_transpose ? 1 : P_1d; 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()
343 const CeedInt stride_0 = is_transpose ? 1 : impl->num_nodes; in CeedBasisApplyNonTensorInterp_Sycl() local
368 V += d_B[i * stride_0 + j * stride_1] * U[j]; in CeedBasisApplyNonTensorInterp_Sycl()
384 const CeedInt stride_0 = is_transpose ? 1 : impl->num_nodes; in CeedBasisApplyNonTensorGrad_Sycl() local
414 const CeedScalar *G = d_G + i * stride_0 + d1 * g_dim_stride; in CeedBasisApplyNonTensorGrad_Sycl()