Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h207 m_planeInputStride = m_inputDepth; in CustomTensorEvaluator()
326 origInputPlane * m_planeInputStride + in coeff()
425 m_planeInputStride * inputPlanes[0] + in packet()
543 inputCoords[1] * m_planeInputStride + inputCoords[0]; in coeff()
548 inputCoords[3] * m_planeInputStride + inputCoords[4]; in coeff()
613 Index m_planeInputStride; member
Deigen_cuboid_convolution.h165 m_planeInputStride = m_inputDepth; in TensorContractionInputMapper()
222 m_planeInputStride = base_mapper.m_planeInputStride; in TensorContractionInputMapper()
330 EIGEN_ALWAYS_INLINE Index patchDepth() const { return m_planeInputStride; } in patchDepth()
390 const Index inputIndex = depth + origInputPlane * m_planeInputStride + in loadCoeff()
425 const Index inputIndex = depth + inputPlane * m_planeInputStride + in loadCoeffStandard()
511 depth + inputPlanes[0] * m_planeInputStride + in loadPacketStandard()
557 const Index inputIndex = depth + inputPlane * m_planeInputStride + in loadPacketFast()
623 Index m_planeInputStride; // Plane stride in the input tensor variable
822 m_base_mapper.m_planeInputStride && in patchDepth()
824 return m_base_mapper.m_planeInputStride; in patchDepth()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h316 m_planeInputStride = m_inputDepth;
402 origInputPlane * m_planeInputStride +
489 m_planeInputStride * inputPlanes[0] +
583 Index m_planeInputStride;