Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions.h102 const StorageIndex peeled_k = (rows / packet_size) * packet_size;
105 const StorageIndex max_col = rhs.maxCol(peeled_k);
112 eigen_assert(k <= peeled_k);
115 const StorageIndex max_row = rhs.maxRow(peeled_k, c);
124 std::min<StorageIndex>(start_depth + (peeled_k - k),
136 eigen_assert(k < peeled_k);
147 eigen_assert(k < peeled_k);
153 eigen_assert(k < peeled_k);
167 eigen_assert(k <= peeled_k);
172 rhs.maxDepth(peeled_k - k, start_depth);
[all …]
Deigen_cuboid_convolution.h781 EIGEN_ALWAYS_INLINE Index maxCol(const Index peeled_k) const { in maxCol() argument
783 fastPatchColStride().divide(m_depth_offset + peeled_k); in maxCol()
788 EIGEN_ALWAYS_INLINE Index maxRow(const Index peeled_k, in maxRow() argument
791 m_depth_offset + peeled_k - col * patchColStride()); in maxRow()
796 EIGEN_ALWAYS_INLINE Index maxPlane(const Index peeled_k, const Index col, in maxPlane() argument
799 m_depth_offset + peeled_k - col * patchColStride() - in maxPlane()
1039 const Index peeled_k = (depth / packet_size) * packet_size;
1055 const Index max_col = rhs.maxCol(peeled_k);
1058 eigen_assert(k <= peeled_k);
1061 const Index max_row = rhs.maxRow(peeled_k, c);
[all …]
Deigen_spatial_convolutions-inl.h612 EIGEN_ALWAYS_INLINE Index maxCol(const Index peeled_k) const { in maxCol() argument
614 (m_depth_offset + (peeled_k == 0 ? 0 : peeled_k - 1)) / in maxCol()
620 EIGEN_ALWAYS_INLINE Index maxRow(const Index peeled_k, in maxRow() argument
622 const Index max_row = (m_depth_offset + (peeled_k == 0 ? 0 : peeled_k - 1) - in maxRow()
629 EIGEN_ALWAYS_INLINE Index maxDepth(const Index peeled_k, const Index col, in maxDepth() argument
631 const Index max_depth = m_depth_offset + peeled_k - // in maxDepth()
840 const Index peeled_k = (depth / packet_size) * packet_size;
856 const Index max_col = rhs.maxCol(peeled_k);
859 eigen_assert(k <= peeled_k);
862 const Index max_row = rhs.maxRow(peeled_k, c);
[all …]
/external/eigen/Eigen/src/Core/products/
DGeneralBlockPanelKernel.h1831 const Index peeled_k = (depth/PacketSize)*PacketSize;
1835 for(; k<peeled_k; k+=PacketSize)
1911 const Index peeled_k = (depth/PacketSize)*PacketSize;
1972 for(; k<peeled_k; k+=PacketSize) {