Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h194 m_rowStride = m_dimensions[1]; in CustomTensorEvaluator()
195 m_colStride = m_dimensions[2] * m_rowStride; in CustomTensorEvaluator()
199 m_rowStride = m_dimensions[NumDims - 2]; in CustomTensorEvaluator()
200 m_colStride = m_dimensions[NumDims - 3] * m_rowStride; in CustomTensorEvaluator()
219 m_fastRowStride = internal::TensorIntDivisor<Index>(m_rowStride); in CustomTensorEvaluator()
304 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in coeff()
404 rowOffsets[0] * m_rowStride, in packet()
406 rowOffsets[1] * m_rowStride}; in packet()
609 Index m_rowStride; member
Deigen_cuboid_convolution.h128 m_rowStride = m_patch_planes; in TensorContractionInputMapper()
129 m_colStride = m_patch_rows * m_rowStride; in TensorContractionInputMapper()
191 m_fastRowStride = internal::TensorIntDivisor<Index>(m_rowStride); in TensorContractionInputMapper()
217 m_rowStride = base_mapper.m_rowStride; in TensorContractionInputMapper()
332 EIGEN_ALWAYS_INLINE Index patchPlanes() const { return m_rowStride; } in patchPlanes()
373 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadCoeff()
412 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadCoeffStandard()
497 rowOffsets[0] * m_rowStride, in loadPacketStandard()
499 rowOffsets[1] * m_rowStride}; in loadPacketStandard()
544 patchOffset - colOffset * m_colStride - rowOffset * m_rowStride; in loadPacketFast()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h304 m_rowStride = m_dimensions[1];
305 m_colStride = m_dimensions[2] * m_rowStride;
309 m_rowStride = m_dimensions[NumDims-2];
310 m_colStride = m_dimensions[NumDims-3] * m_rowStride;
327 m_fastRowStride = internal::TensorIntDivisor<Index>(m_rowStride);
388 const Index planeOffset = patchOffset - colOffset * m_colStride - rowOffset * m_rowStride;
471 patchOffsets[0] - colOffsets[0] * m_colStride - rowOffsets[0] * m_rowStride,
472 patchOffsets[1] - colOffsets[1] * m_colStride - rowOffsets[1] * m_rowStride};
579 Index m_rowStride;