tesseract::StrideMap Class Reference

#include <stridemap.h>

Classes
class	Index

Public Member Functions
	StrideMap ()
void	SetStride (const std::vector< std::pair< int, int >> &h_w_pairs)
void	ScaleXY (int x_factor, int y_factor)
void	ReduceWidthTo1 ()
void	TransposeXY ()
int	Size (FlexDimensions dimension) const
int	Width () const

Detailed Description

Definition at line 41 of file stridemap.h.

Constructor & Destructor Documentation

StrideMap()

tesseract::StrideMap::StrideMap ( )

inline

Definition at line 99 of file stridemap.h.

              {
    memset(shape_, 0, sizeof(shape_));
    memset(t_increments_, 0, sizeof(t_increments_));
  }

Member Function Documentation

ReduceWidthTo1()

void tesseract::StrideMap::ReduceWidthTo1

(

)

Definition at line 153 of file stridemap.cpp.

                               {
  widths_.assign(widths_.size(), 1);
  shape_[FD_WIDTH] = 1;
  ComputeTIncrements();
}

ScaleXY()

void tesseract::StrideMap::ScaleXY	(	int	x_factor,
		int	y_factor
	)

Definition at line 144 of file stridemap.cpp.

                                                  {
  for (int& height : heights_) height /= y_factor;
  for (int& width : widths_) width /= x_factor;
  shape_[FD_HEIGHT] /= y_factor;
  shape_[FD_WIDTH] /= x_factor;
  ComputeTIncrements();
}

SetStride()

void tesseract::StrideMap::SetStride

(

const std::vector< std::pair< int, int >> &

h_w_pairs

)

Definition at line 126 of file stridemap.cpp.

                                                                       {
  int max_height = 0;
  int max_width = 0;
  for (const std::pair<int, int>& hw : h_w_pairs) {
    int height = hw.first;
    int width = hw.second;
    heights_.push_back(height);
    widths_.push_back(width);
    if (height > max_height) max_height = height;
    if (width > max_width) max_width = width;
  }
  shape_[FD_BATCH] = heights_.size();
  shape_[FD_HEIGHT] = max_height;
  shape_[FD_WIDTH] = max_width;
  ComputeTIncrements();
}

Size()

int tesseract::StrideMap::Size ( FlexDimensions  dimension ) const

inline

Definition at line 114 of file stridemap.h.

{ return shape_[dimension]; }

TransposeXY()

void tesseract::StrideMap::TransposeXY

(

)

Definition at line 160 of file stridemap.cpp.

                            {
  std::swap(shape_[FD_HEIGHT], shape_[FD_WIDTH]);
  std::swap(heights_, widths_);
  ComputeTIncrements();
}

Width()

int tesseract::StrideMap::Width ( ) const

inline

Definition at line 116 of file stridemap.h.

{ return t_increments_[FD_BATCH] * shape_[FD_BATCH]; }

---

The documentation for this class was generated from the following files:

• /tesseract/src/lstm/stridemap.h
• /tesseract/src/lstm/stridemap.cpp