Searched refs:base_mapper (Results 1 – 3 of 3) sorted by relevance
205 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() argument206 : m_impl(base_mapper.m_impl) { in TensorContractionInputMapper()207 m_patch_depth = base_mapper.m_patch_depth; in TensorContractionInputMapper()208 m_patch_planes = base_mapper.m_patch_planes; in TensorContractionInputMapper()209 m_patch_rows = base_mapper.m_patch_rows; in TensorContractionInputMapper()210 m_patch_cols = base_mapper.m_patch_cols; in TensorContractionInputMapper()211 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()213 m_patch_plane_stride = base_mapper.m_patch_plane_stride; in TensorContractionInputMapper()214 m_patch_row_stride = base_mapper.m_patch_row_stride; in TensorContractionInputMapper()215 m_patch_col_stride = base_mapper.m_patch_col_stride; in TensorContractionInputMapper()[all …]
165 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() argument166 : m_impl(base_mapper.m_impl) { in TensorContractionInputMapper()167 m_patch_cols = base_mapper.m_patch_cols; in TensorContractionInputMapper()168 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()170 m_patch_row_stride = base_mapper.m_patch_row_stride; in TensorContractionInputMapper()171 m_patch_col_stride = base_mapper.m_patch_col_stride; in TensorContractionInputMapper()173 m_patch_row_inflate_strides = base_mapper.m_patch_row_inflate_strides; in TensorContractionInputMapper()174 m_patch_col_inflate_strides = base_mapper.m_patch_col_inflate_strides; in TensorContractionInputMapper()176 m_colStride = base_mapper.m_colStride; in TensorContractionInputMapper()178 m_rowInputStride = base_mapper.m_rowInputStride; in TensorContractionInputMapper()[all …]
352 …EIGEN_DEVICE_FUNC TensorContractionSubMapper(const ParentMapper& base_mapper, Index vert_offset, I…353 : m_base_mapper(base_mapper), m_vert_offset(vert_offset), m_horiz_offset(horiz_offset) {