Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionCuda.h576 Index rhs_horiz0 = (threadIdx.x>>2)+threadIdx.y*4+base_n; in EigenFloatContractionKernelInternal16x16() local
581 rhs_pf0 = rhs.loadPacket<Unaligned>(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
584 rhs_pf0.x = rhs(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
585 rhs_pf0.y = rhs(rhs_vert + 1, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
586 rhs_pf0.z = rhs(rhs_vert + 2, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
588 rhs_pf0.x = rhs(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
589 rhs_pf0.y = rhs(rhs_vert + 1, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
591 rhs_pf0.x = rhs(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
594 if (rhs_horiz0 < n_size) { in EigenFloatContractionKernelInternal16x16()
596 rhs_pf0 = rhs.loadPacket<Unaligned>(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
[all …]