Searched refs:m_col_offset (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_spatial_convolutions-inl.h | 718 m_col_offset(horiz_offset), in TensorContractionSubMapper() 720 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper() 726 m_col_offset(horiz_offset + base_mapper.m_col_offset), in TensorContractionSubMapper() 728 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper() 737 return m_base_mapper(i + m_depth_offset, j + m_col_offset); in operator() 747 j + m_col_offset); in loadPacket() 991 return LinearMapper(m_base_mapper, i + m_depth_offset, j + m_col_offset); in getLinearMapper() 996 Index m_col_offset; // First col in the input matrix variable
|
D | eigen_cuboid_convolution.h | 913 m_col_offset(horiz_offset) { in TensorContractionSubMapper() 914 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper() 921 m_col_offset(horiz_offset + base_mapper.m_col_offset) { in TensorContractionSubMapper() 922 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper() 931 return m_base_mapper(i + m_depth_offset, j + m_col_offset); in operator() 941 j + m_col_offset); in loadPacket() 1142 return LinearMapper(m_base_mapper, i + m_depth_offset, j + m_col_offset); in getLinearMapper() 1150 Index m_col_offset; // First col in the input matrix variable
|