tesseract::NetworkScratch::IO Class Reference

#include <networkscratch.h>

Public Member Functions
	IO (const NetworkIO &src, NetworkScratch *scratch)
	IO ()
	~IO ()
void	Resize (const NetworkIO &src, int num_features, NetworkScratch *scratch)
void	Resize2d (bool int_mode, int width, int num_features, NetworkScratch *scratch)
void	ResizeFloat (const NetworkIO &src, int num_features, NetworkScratch *scratch)
NetworkIO &	operator* ()
NetworkIO *	operator-> ()
	operator NetworkIO * ()

Detailed Description

Definition at line 51 of file networkscratch.h.

Constructor & Destructor Documentation

IO() [1/2]

tesseract::NetworkScratch::IO::IO ( const NetworkIO &  src, NetworkScratch *  scratch  )

inline

Definition at line 54 of file networkscratch.h.

      : int_mode_(scratch->int_mode_ && src.int_mode()),
        scratch_space_(scratch) {
      network_io_ = int_mode_ ? scratch_space_->int_stack_.Borrow()
                              : scratch_space_->float_stack_.Borrow();
    }

IO() [2/2]

tesseract::NetworkScratch::IO::IO ( )

inline

Definition at line 62 of file networkscratch.h.

: int_mode_(false), network_io_(nullptr), scratch_space_(nullptr) {}

~IO()

tesseract::NetworkScratch::IO::~IO ( )

inline

Definition at line 64 of file networkscratch.h.

          {
      if (scratch_space_ == nullptr) {
        ASSERT_HOST(network_io_ == nullptr);
      } else if (int_mode_) {
        scratch_space_->int_stack_.Return(network_io_);
      } else {
        scratch_space_->float_stack_.Return(network_io_);
      }
    }

Member Function Documentation

operator NetworkIO *()

tesseract::NetworkScratch::IO::operator NetworkIO * ( )

inline

Definition at line 117 of file networkscratch.h.

                          {
      return network_io_;
    }

operator*()

NetworkIO& tesseract::NetworkScratch::IO::operator* ( )

inline

Definition at line 111 of file networkscratch.h.

                           {
      return *network_io_;
    }

operator->()

NetworkIO* tesseract::NetworkScratch::IO::operator-> ( )

inline

Definition at line 114 of file networkscratch.h.

                            {
      return network_io_;
    }

Resize()

void tesseract::NetworkScratch::IO::Resize ( const NetworkIO &  src, int  num_features, NetworkScratch *  scratch  )

inline

Definition at line 76 of file networkscratch.h.

                                         {
      if (scratch_space_ == nullptr) {
        int_mode_ = scratch->int_mode_ && src.int_mode();
        scratch_space_ = scratch;
        network_io_ = int_mode_ ? scratch_space_->int_stack_.Borrow()
                                : scratch_space_->float_stack_.Borrow();
      }
      network_io_->Resize(src, num_features);
    }

Resize2d()

void tesseract::NetworkScratch::IO::Resize2d ( bool  int_mode, int  width, int  num_features, NetworkScratch *  scratch  )

inline

Definition at line 87 of file networkscratch.h.

                                           {
      if (scratch_space_ == nullptr) {
        int_mode_ = scratch->int_mode_ && int_mode;
        scratch_space_ = scratch;
        network_io_ = int_mode_ ? scratch_space_->int_stack_.Borrow()
                                : scratch_space_->float_stack_.Borrow();
      }
      network_io_->Resize2d(int_mode, width, num_features);
    }

ResizeFloat()

void tesseract::NetworkScratch::IO::ResizeFloat ( const NetworkIO &  src, int  num_features, NetworkScratch *  scratch  )

inline

Definition at line 99 of file networkscratch.h.

                                              {
      if (scratch_space_ == nullptr) {
        int_mode_ = false;
        scratch_space_ = scratch;
        network_io_ = scratch_space_->float_stack_.Borrow();
      }
      network_io_->ResizeFloat(src, num_features);
    }

---

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

• /tesseract/src/lstm/networkscratch.h