Searched refs:base_mapper (Results 1 – 4 of 4) sorted by relevance
207 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() argument208 : m_impl(base_mapper.m_impl) { in TensorContractionInputMapper()209 m_patch_depth = base_mapper.m_patch_depth; in TensorContractionInputMapper()210 m_patch_planes = base_mapper.m_patch_planes; in TensorContractionInputMapper()211 m_patch_rows = base_mapper.m_patch_rows; in TensorContractionInputMapper()212 m_patch_cols = base_mapper.m_patch_cols; in TensorContractionInputMapper()213 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()215 m_patch_plane_stride = base_mapper.m_patch_plane_stride; in TensorContractionInputMapper()216 m_patch_row_stride = base_mapper.m_patch_row_stride; in TensorContractionInputMapper()217 m_patch_col_stride = base_mapper.m_patch_col_stride; in TensorContractionInputMapper()[all …]
170 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() argument171 : m_impl(base_mapper.m_impl) { in TensorContractionInputMapper()172 m_patch_cols = base_mapper.m_patch_cols; in TensorContractionInputMapper()173 m_num_patches = base_mapper.m_num_patches; in TensorContractionInputMapper()175 m_patch_row_stride = base_mapper.m_patch_row_stride; in TensorContractionInputMapper()176 m_patch_col_stride = base_mapper.m_patch_col_stride; in TensorContractionInputMapper()178 m_patch_row_inflate_strides = base_mapper.m_patch_row_inflate_strides; in TensorContractionInputMapper()179 m_patch_col_inflate_strides = base_mapper.m_patch_col_inflate_strides; in TensorContractionInputMapper()181 m_colStride = base_mapper.m_colStride; in TensorContractionInputMapper()183 m_rowInputStride = base_mapper.m_rowInputStride; in TensorContractionInputMapper()[all …]
443 Side == Lhs ? data_mapper.base_mapper().stride() \444 : data_mapper.base_mapper().nocontract_strides()[0]; \445 const Scalar* data = data_mapper.base_mapper().tensor().data(); \
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) {