#include <stridemap.h>
Definition at line 44 of file stridemap.h.
◆ Index() [1/2]
tesseract::StrideMap::Index::Index |
( |
const StrideMap & |
stride_map | ) |
|
|
inlineexplicit |
Definition at line 46 of file stridemap.h.
46 : stride_map_(&stride_map) {
48 }
◆ Index() [2/2]
tesseract::StrideMap::Index::Index |
( |
const StrideMap & |
stride_map, |
|
|
int |
batch, |
|
|
int |
y, |
|
|
int |
x |
|
) |
| |
|
inline |
Definition at line 49 of file stridemap.h.
50 : stride_map_(&stride_map) {
54 SetTFromIndices();
55 }
◆ AddOffset()
bool tesseract::StrideMap::Index::AddOffset |
( |
int |
offset, |
|
|
FlexDimensions |
dimension |
|
) |
| |
Definition at line 62 of file stridemap.cpp.
62 {
63 indices_[dimension] += offset;
64 SetTFromIndices();
66}
◆ Decrement()
bool tesseract::StrideMap::Index::Decrement |
( |
| ) |
|
Definition at line 87 of file stridemap.cpp.
87 {
89 if (indices_[d] > 0) {
90 --indices_[d];
92
93
94 InitToLastOfBatch(indices_[
FD_BATCH]);
95 } else {
96 t_ -= stride_map_->t_increments_[d];
97 }
98 return true;
99 }
101 t_ += stride_map_->t_increments_[d] * indices_[d];
102
103 }
104 return false;
105}
int MaxIndexOfDim(FlexDimensions dim) const
◆ Increment()
bool tesseract::StrideMap::Index::Increment |
( |
| ) |
|
Definition at line 70 of file stridemap.cpp.
70 {
73 t_ += stride_map_->t_increments_[d];
74 ++indices_[d];
75 return true;
76 }
77 t_ -= stride_map_->t_increments_[d] * indices_[d];
78 indices_[d] = 0;
79
80 }
81 return false;
82}
bool IsLast(FlexDimensions dimension) const
◆ index()
int tesseract::StrideMap::Index::index |
( |
FlexDimensions |
dimension | ) |
const |
|
inline |
Definition at line 58 of file stridemap.h.
58{ return indices_[dimension]; }
◆ InitToFirst()
void tesseract::StrideMap::Index::InitToFirst |
( |
| ) |
|
|
inline |
Definition at line 60 of file stridemap.h.
60 {
61 memset(indices_, 0, sizeof(indices_));
62 t_ = 0;
63 }
◆ InitToLast()
void tesseract::StrideMap::Index::InitToLast |
( |
| ) |
|
|
inline |
◆ IsLast()
bool tesseract::StrideMap::Index::IsLast |
( |
FlexDimensions |
dimension | ) |
const |
◆ IsValid()
bool tesseract::StrideMap::Index::IsValid |
( |
| ) |
const |
Definition at line 24 of file stridemap.cpp.
24 {
25
26 for (
int index : indices_) {
27 if (
index < 0)
return false;
28 }
31 return false;
32 }
33 return true;
34}
int index(FlexDimensions dimension) const
◆ MaxIndexOfDim()
int tesseract::StrideMap::Index::MaxIndexOfDim |
( |
FlexDimensions |
dim | ) |
const |
Definition at line 43 of file stridemap.cpp.
43 {
44 int max_index = stride_map_->shape_[dim] - 1;
45 if (dim ==
FD_BATCH)
return max_index;
47 const size_t batch = indices_[
FD_BATCH];
49 if (batch >= stride_map_->heights_.size() ||
50 stride_map_->heights_[batch] > max_index)
51 return max_index;
52 return stride_map_->heights_[batch] - 1;
53 }
54 if (batch >= stride_map_->widths_.size() ||
55 stride_map_->widths_[batch] > max_index)
56 return max_index;
57 return stride_map_->widths_[batch] - 1;
58}
◆ t()
int tesseract::StrideMap::Index::t |
( |
| ) |
const |
|
inline |
The documentation for this class was generated from the following files: