Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions-inl.h108 m_patch_cols = tensor.impl().dimensions()[2]; in TensorContractionInputMapper()
114 m_patch_cols = tensor.impl().dimensions()[NumDims - 3]; in TensorContractionInputMapper()
167 m_patch_cols = base_mapper.m_patch_cols; in TensorContractionInputMapper()
268 EIGEN_ALWAYS_INLINE Index patchCols() const { return m_patch_cols; } in patchCols()
345 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacket()
359 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketStandard()
412 eigen_assert(patchId < patchDepth() * patchRows() * m_patch_cols); in loadPacketFast()
464 Index m_patch_cols; // number of columns in the patch variable
662 return m_base_mapper.m_patch_cols; in patchCols()
Deigen_cuboid_convolution.h108 m_patch_cols = tensor.impl().dimensions()[3]; in TensorContractionInputMapper()
115 m_patch_cols = tensor.impl().dimensions()[NumDims - 4]; in TensorContractionInputMapper()
130 m_patchStride = m_colStride * m_patch_cols * m_patch_depth; in TensorContractionInputMapper()
210 m_patch_cols = base_mapper.m_patch_cols; in TensorContractionInputMapper()
336 EIGEN_ALWAYS_INLINE Index patchCols() const { return m_patch_cols; } in patchCols()
609 Index m_patch_cols; // number of columns in the patch variable
839 return m_base_mapper.m_patch_cols; in patchCols()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h73 : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
87 : m_xpr(expr), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
98 DenseIndex patch_cols() const { return m_patch_cols; }
133 const DenseIndex m_patch_cols;
DTensorVolumePatch.h68 … : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
83 … : m_xpr(expr), m_patch_planes(patch_planes), m_patch_rows(patch_rows), m_patch_cols(patch_cols),
96 DenseIndex patch_cols() const { return m_patch_cols; }
142 const DenseIndex m_patch_cols;