Home
last modified time | relevance | path

Searched refs:m_depth_offset (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions-inl.h717 : m_depth_offset(vert_offset), in TensorContractionSubMapper()
725 : m_depth_offset(vert_offset + base_mapper.m_depth_offset), in TensorContractionSubMapper()
732 return m_base_mapper.loadCoeff(i + m_depth_offset, m_rowIndex, m_colIndex, in operator()
737 return m_base_mapper(i + m_depth_offset, j + m_col_offset); in operator()
741 return m_base_mapper.loadPacket(i + m_depth_offset, m_rowIndex, m_colIndex, in loadPacket()
746 return m_base_mapper.template loadPacket<Alignment>(i + m_depth_offset, in loadPacket()
751 return m_base_mapper.loadCoeffStandard(i + m_depth_offset, m_rowIndex, in loadCoeffStandard()
756 return m_base_mapper.loadPacketFast(i + m_depth_offset, m_rowIndex, in loadPacketFast()
763 i + m_depth_offset, m_rowIndex, m_colIndex, m_otherIndex); in loadPacketStandard()
782 (m_depth_offset + (peeled_k == 0 ? 0 : peeled_k - 1)) / in maxCol()
[all …]
Deigen_cuboid_convolution.h912 m_depth_offset(vert_offset), in TensorContractionSubMapper()
920 m_depth_offset(vert_offset + base_mapper.m_depth_offset), in TensorContractionSubMapper()
926 return m_base_mapper.loadCoeff(i + m_depth_offset, m_planeIndex, m_rowIndex, in operator()
931 return m_base_mapper(i + m_depth_offset, j + m_col_offset); in operator()
935 return m_base_mapper.loadPacket(i + m_depth_offset, m_planeIndex, in loadPacket()
940 return m_base_mapper.template loadPacket<Alignment>(i + m_depth_offset, in loadPacket()
947 i + m_depth_offset, m_planeIndex, m_rowIndex, m_colIndex, m_otherIndex); in loadCoeffStandard()
951 return m_base_mapper.loadPacketFast(i + m_depth_offset, m_planeIndex, in loadPacketFast()
958 i + m_depth_offset, m_planeIndex, m_rowIndex, m_colIndex, m_otherIndex); in loadPacketStandard()
977 fastPatchColStride().divide(m_depth_offset + peeled_k); in maxCol()
[all …]