Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionMapper.h240 EIGEN_STRONG_INLINE Packet loadPacket(Index i, Index j) const {
289 return loadPacket<AlignmentType>(i, j);
320 EIGEN_STRONG_INLINE Packet loadPacket(Index i, Index j) const {
327 return loadPacket(i, j);
375 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i) const {
377 return m_base_mapper.template loadPacket<Alignment>(i, 0);
379 return m_base_mapper.template loadPacket<Alignment>(i + m_vert_offset, m_horiz_offset);
381 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i, Index j) const {
383 return m_base_mapper.template loadPacket<Alignment>(i, j);
385 return m_base_mapper.template loadPacket<Alignment>(i + m_vert_offset, j + m_horiz_offset);
[all …]
DTensorContractionCuda.h546 reg =lhs.loadPacket<Unaligned>(row, col); \ in EigenFloatContractionKernelInternal16x16()
551 reg =lhs.loadPacket<Unaligned>(row, col); \ in EigenFloatContractionKernelInternal16x16()
581 rhs_pf0 = rhs.loadPacket<Unaligned>(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
596 rhs_pf0 = rhs.loadPacket<Unaligned>(rhs_vert, rhs_horiz0); in EigenFloatContractionKernelInternal16x16()
793 lhs_pf0 =lhs.loadPacket<Unaligned>(lhs_vert, (threadIdx.y/4+k)); in EigenFloatContractionKernelInternal()
794 lhs_pf1 =lhs.loadPacket<Unaligned>(lhs_vert, (threadIdx.y/4+k+8)); in EigenFloatContractionKernelInternal()
795 lhs_pf2 =lhs.loadPacket<Unaligned>(lhs_vert, (threadIdx.y/4+k+16)); in EigenFloatContractionKernelInternal()
796 lhs_pf3 =lhs.loadPacket<Unaligned>(lhs_vert, (threadIdx.y/4+k+24)); in EigenFloatContractionKernelInternal()
798 lhs_pf0 =lhs.loadPacket<Unaligned>(lhs_vert, (threadIdx.y/4+k)); in EigenFloatContractionKernelInternal()
799 lhs_pf1 =lhs.loadPacket<Unaligned>(lhs_vert, (threadIdx.y/4+k+8)); in EigenFloatContractionKernelInternal()
[all …]
/external/eigen/Eigen/src/Core/products/
DGeneralBlockPanelKernel.h1028 R0 = r0.loadPacket(0 * Traits::ResPacketSize);
1029 R1 = r0.loadPacket(1 * Traits::ResPacketSize);
1030 R2 = r0.loadPacket(2 * Traits::ResPacketSize);
1038 R0 = r1.loadPacket(0 * Traits::ResPacketSize);
1039 R1 = r1.loadPacket(1 * Traits::ResPacketSize);
1040 R2 = r1.loadPacket(2 * Traits::ResPacketSize);
1048 R0 = r2.loadPacket(0 * Traits::ResPacketSize);
1049 R1 = r2.loadPacket(1 * Traits::ResPacketSize);
1050 R2 = r2.loadPacket(2 * Traits::ResPacketSize);
1058 R0 = r3.loadPacket(0 * Traits::ResPacketSize);
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_contraction_kernel.h94 const Packet p = lm.template loadPacket<Packet>(row + j * PacketSize);
101 const Packet p = lm.template loadPacket<Packet>(row);
Deigen_spatial_convolutions-inl.h243 EIGEN_ALWAYS_INLINE Packet loadPacket(Index row) const { in loadPacket() function
246 return loadPacket(row, rowIndex, colIndex, otherIndex); in loadPacket()
252 EIGEN_ALWAYS_INLINE Packet loadPacket(Index row, Index patchIndex) const { in loadPacket() function
255 return loadPacket(row, rowIndex, colIndex, otherIndex); in loadPacket()
340 EIGEN_ALWAYS_INLINE Packet loadPacket(Index patchId, Index rowIndex, in loadPacket() function
573 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i) const { in loadPacket() function
574 return m_base_mapper.loadPacket(i + m_depth_offset, m_rowIndex, m_colIndex, in loadPacket()
577 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i, in loadPacket() function
579 return m_base_mapper.template loadPacket<Alignment>(i + m_depth_offset, in loadPacket()
Deigen_cuboid_convolution.h309 EIGEN_ALWAYS_INLINE Packet loadPacket(Index row) const { in loadPacket() function
312 return loadPacket(row, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
318 EIGEN_ALWAYS_INLINE Packet loadPacket(Index row, Index patchIndex) const { in loadPacket() function
321 return loadPacket(row, planeIndex, rowIndex, colIndex, otherIndex); in loadPacket()
435 EIGEN_ALWAYS_INLINE Packet loadPacket(Index patchId, Index planeIndex, in loadPacket() function
741 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i) const { in loadPacket() function
742 return m_base_mapper.loadPacket(i + m_depth_offset, m_planeIndex, in loadPacket()
745 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i, in loadPacket() function
747 return m_base_mapper.template loadPacket<Alignment>(i + m_depth_offset, in loadPacket()
/external/eigen/Eigen/src/Core/util/
DBlasUtil.h174 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i) const {
221 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet loadPacket(Index i, Index j) const {