Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionCuda.h908 Index rhs_horiz1 = threadIdx.y*2+1+base_n; in EigenFloatContractionKernelInternal() local
913 rhs_pf1 = rhs.loadPacket<Unaligned>(rhs_vert, rhs_horiz1); in EigenFloatContractionKernelInternal()
919 rhs_pf1.x = rhs(rhs_vert, rhs_horiz1); in EigenFloatContractionKernelInternal()
920 rhs_pf1.y = rhs(rhs_vert + 1, rhs_horiz1); in EigenFloatContractionKernelInternal()
921 rhs_pf1.z = rhs(rhs_vert + 2, rhs_horiz1); in EigenFloatContractionKernelInternal()
925 rhs_pf1.x = rhs(rhs_vert, rhs_horiz1); in EigenFloatContractionKernelInternal()
926 rhs_pf1.y = rhs(rhs_vert + 1, rhs_horiz1); in EigenFloatContractionKernelInternal()
929 rhs_pf1.x = rhs(rhs_vert, rhs_horiz1); in EigenFloatContractionKernelInternal()
932 if (rhs_horiz1 < n_size) { in EigenFloatContractionKernelInternal()
936 rhs_pf1 = rhs.loadPacket<Unaligned>(rhs_vert, rhs_horiz1); in EigenFloatContractionKernelInternal()
[all …]