Home
last modified time | relevance | path

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

/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 …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_cuboid_convolution.h718 : m_base_mapper(base_mapper), in TensorContractionSubMapper()
721 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
726 : m_base_mapper(base_mapper.m_base_mapper), in TensorContractionSubMapper()
729 m_base_mapper.computeBaseIndices(m_col_offset, m_planeIndex, m_rowIndex, in TensorContractionSubMapper()
733 return m_base_mapper.loadCoeff(i + m_depth_offset, m_planeIndex, m_rowIndex, in operator()
738 return m_base_mapper(i + m_depth_offset, j + m_col_offset); in operator()
742 return m_base_mapper.loadPacket(i + m_depth_offset, m_planeIndex, in loadPacket()
747 return m_base_mapper.template loadPacket<Alignment>(i + m_depth_offset, in loadPacket()
753 return m_base_mapper.loadCoeffStandard( in loadCoeffStandard()
758 return m_base_mapper.loadPacketFast(i + m_depth_offset, m_planeIndex, in loadPacketFast()
[all …]
Deigen_spatial_convolutions-inl.h552 m_base_mapper(base_mapper) { in TensorContractionSubMapper()
553 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
560 m_base_mapper(base_mapper.m_base_mapper) { in TensorContractionSubMapper()
561 m_base_mapper.computeBaseIndices(m_col_offset, m_rowIndex, m_colIndex, in TensorContractionSubMapper()
565 return m_base_mapper.loadCoeff(i + m_depth_offset, m_rowIndex, m_colIndex, in operator()
570 return m_base_mapper(i + m_depth_offset, j + m_col_offset); in operator()
574 return m_base_mapper.loadPacket(i + m_depth_offset, m_rowIndex, m_colIndex, in loadPacket()
579 return m_base_mapper.template loadPacket<Alignment>(i + m_depth_offset, in loadPacket()
584 return m_base_mapper.loadCoeffStandard(i + m_depth_offset, m_rowIndex, in loadCoeffStandard()
589 return m_base_mapper.loadPacketFast(i + m_depth_offset, m_rowIndex, in loadPacketFast()
[all …]