Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions-inl.h719 m_base_mapper(base_mapper) { in TensorContractionSubMapper()
720 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
727 m_base_mapper(base_mapper.m_base_mapper) { in TensorContractionSubMapper()
728 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, 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()
[all …]
Deigen_cuboid_convolution.h911 : m_base_mapper(base_mapper), in TensorContractionSubMapper()
914 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
919 : m_base_mapper(base_mapper.m_base_mapper), in TensorContractionSubMapper()
922 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, 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()
946 return m_base_mapper.loadCoeffStandard( in loadCoeffStandard()
951 return m_base_mapper.loadPacketFast(i + m_depth_offset, m_planeIndex, in loadPacketFast()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionMapper.h353 : m_base_mapper(base_mapper), m_vert_offset(vert_offset), m_horiz_offset(horiz_offset) {
357 Index stride = m_base_mapper.stride();
358 m_base_mapper.offsetBuffer(vert_offset + horiz_offset * stride);
364 return m_base_mapper(i, 0);
366 return m_base_mapper(i + m_vert_offset, m_horiz_offset);
370 return m_base_mapper(i, j);
372 return m_base_mapper(i + m_vert_offset, j + m_horiz_offset);
377 return m_base_mapper.template loadPacket<Alignment>(i, 0);
379 return m_base_mapper.template loadPacket<Alignment>(i + m_vert_offset, m_horiz_offset);
383 return m_base_mapper.template loadPacket<Alignment>(i, j);
[all …]