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.h179 const StorageIndex peeled_k = (rows / packet_size) * packet_size;
182 const StorageIndex max_col = rhs.maxCol(peeled_k);
190 eigen_assert(k <= peeled_k);
193 const StorageIndex max_row = rhs.maxRow(peeled_k, c);
206 std::min<StorageIndex>(start_depth + (peeled_k - k),
219 eigen_assert(k < peeled_k);
236 eigen_assert(k < peeled_k);
247 eigen_assert(k < peeled_k);
265 eigen_assert(k < peeled_k);
271 eigen_assert(k <= peeled_k);
[all …]
Deigen_cuboid_convolution.h975 EIGEN_ALWAYS_INLINE Index maxCol(const Index peeled_k) const { in maxCol() argument
977 fastPatchColStride().divide(m_depth_offset + peeled_k); in maxCol()
982 EIGEN_ALWAYS_INLINE Index maxRow(const Index peeled_k, in maxRow() argument
985 m_depth_offset + peeled_k - col * patchColStride()); in maxRow()
990 EIGEN_ALWAYS_INLINE Index maxPlane(const Index peeled_k, const Index col, in maxPlane() argument
993 m_depth_offset + peeled_k - col * patchColStride() - in maxPlane()
1233 const Index peeled_k = (depth / packet_size) * packet_size;
1249 const Index max_col = rhs.maxCol(peeled_k);
1252 eigen_assert(k <= peeled_k);
1255 const Index max_row = rhs.maxRow(peeled_k, c);
[all …]
Deigen_spatial_convolutions-inl.h780 EIGEN_ALWAYS_INLINE Index maxCol(const Index peeled_k) const { in maxCol() argument
782 (m_depth_offset + (peeled_k == 0 ? 0 : peeled_k - 1)) / in maxCol()
788 EIGEN_ALWAYS_INLINE Index maxRow(const Index peeled_k, in maxRow() argument
790 const Index max_row = (m_depth_offset + (peeled_k == 0 ? 0 : peeled_k - 1) - in maxRow()
797 EIGEN_ALWAYS_INLINE Index maxDepth(const Index peeled_k, const Index col, in maxDepth() argument
799 const Index max_depth = m_depth_offset + peeled_k - // in maxDepth()
1077 const Index peeled_k = (depth / packet_size) * packet_size;
1093 const Index max_col = rhs.maxCol(peeled_k);
1096 eigen_assert(k <= peeled_k);
1099 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) {