LIMA
Libre Multilingual Analyzer — C++ API
Loading...
Searching...
No Matches
Public Types | Public Member Functions | Protected Types | Protected Member Functions | Protected Attributes | List of all members
deeplima::eigen_impl::BiRnnInferenceBase Class Referenceabstract

#include <deeplima/eigen_wrp/birnn_inference_base.h>

Inheritance diagram for deeplima::eigen_impl::BiRnnInferenceBase:
deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation deeplima::tagging::eigen_impl::BiRnnEigenInferenceForTagging< AuxScalar > deeplima::RnnSeq2Seq deeplima::lemmatization::impl::LemmatizationImpl

Public Types

typedef Eigen::MatrixXf Matrix
 
typedef Eigen::VectorXf Vector
 
typedef float Scalar
 
typedef Eigen::MatrixXf tensor_t
 
typedef EmbdUInt64FloatHolder uint_dicts_holder_t
 
typedef EmbdStrFloatHolder str_dicts_holder_t
 

Public Member Functions

virtual ~BiRnnInferenceBase ()=default
 
virtual void load (const std::string &fn)=0
 
const uint_dicts_holder_t & get_input_uint_dicts () const
 
const std::vector< std::string > & get_input_uint_dicts_names () const
 
const str_dicts_holder_t & get_input_str_dicts () const
 
const std::vector< std::string > & get_input_str_dicts_names () const
 
const std::vector< std::vector< std::string > > & get_output_str_dicts () const
 
const std::vector< std::string > & get_output_str_dicts_names () const
 
virtual size_t init_new_worker (size_t input_len, bool precomputed_input=false)
 
virtual void precompute_inputs (const Eigen::MatrixXf &inputs, Eigen::MatrixXf &outputs, int64_t input_size)=0
 
virtual void predict (size_t worker_id, const Eigen::MatrixXf &inputs, int64_t input_begin, int64_t input_end, int64_t output_begin, int64_t output_end, std::shared_ptr< StdMatrix< uint8_t > > &output, const std::vector< std::string > &outputs_names)=0
 

Protected Types

typedef params_bilstm_t< Eigen::MatrixXf, Eigen::VectorXf > params_bilstm_spec_t
 
typedef params_multilayer_bilstm_t< Eigen::MatrixXf, Eigen::VectorXf > params_multilayer_bilstm_spec_t
 

Protected Member Functions

virtual void convert_from_torch (const std::string &fn)=0
 
virtual void convert_dicts_and_embeddings (const nets::BiRnnClassifierImpl &src)
 

Protected Attributes

std::vector< std::shared_ptr< Op_Base > > m_ops
 
std::vector< std::shared_ptr< param_base_t > > m_params
 
std::vector< std::vector< std::shared_ptr< Op_Base::workbench_t > > > m_wb
 
uint_dicts_holder_t m_input_uint_dicts
 
std::vector< std::string > m_input_uint_dicts_names
 
str_dicts_holder_t m_input_str_dicts
 
std::vector< std::string > m_input_str_dicts_names
 
std::vector< std::vector< std::string > > m_output_str_dicts
 
std::vector< std::string > m_output_str_dicts_names
 
std::vector< params_bilstm_spec_t > m_lstm
 
std::map< std::string, size_t > m_lstm_idx
 
std::vector< std::shared_ptr< params_multilayer_bilstm_spec_t > > m_multi_bilstm
 
std::map< std::string, size_t > m_multi_bilstm_idx
 
std::vector< params_linear_t< Eigen::MatrixXf, Eigen::VectorXf > > m_linear
 
std::map< std::string, size_t > m_linear_idx
 

Detailed Description

Definition at line 23 of file birnn_inference_base.h.

Member Typedef Documentation

◆ Matrix

typedef Eigen::MatrixXf deeplima::eigen_impl::BiRnnInferenceBase::Matrix

Definition at line 26 of file birnn_inference_base.h.

◆ params_bilstm_spec_t

typedef params_bilstm_t<Eigen::MatrixXf, Eigen::VectorXf> deeplima::eigen_impl::BiRnnInferenceBase::params_bilstm_spec_t
protected

Definition at line 115 of file birnn_inference_base.h.

◆ params_multilayer_bilstm_spec_t

typedef params_multilayer_bilstm_t<Eigen::MatrixXf, Eigen::VectorXf> deeplima::eigen_impl::BiRnnInferenceBase::params_multilayer_bilstm_spec_t
protected

Definition at line 119 of file birnn_inference_base.h.

◆ Scalar

typedef float deeplima::eigen_impl::BiRnnInferenceBase::Scalar

Definition at line 28 of file birnn_inference_base.h.

◆ str_dicts_holder_t

typedef EmbdStrFloatHolder deeplima::eigen_impl::BiRnnInferenceBase::str_dicts_holder_t

Definition at line 31 of file birnn_inference_base.h.

◆ tensor_t

typedef Eigen::MatrixXf deeplima::eigen_impl::BiRnnInferenceBase::tensor_t

Definition at line 29 of file birnn_inference_base.h.

◆ uint_dicts_holder_t

typedef EmbdUInt64FloatHolder deeplima::eigen_impl::BiRnnInferenceBase::uint_dicts_holder_t

Definition at line 30 of file birnn_inference_base.h.

◆ Vector

typedef Eigen::VectorXf deeplima::eigen_impl::BiRnnInferenceBase::Vector

Definition at line 27 of file birnn_inference_base.h.

Constructor & Destructor Documentation

◆ ~BiRnnInferenceBase()

virtual deeplima::eigen_impl::BiRnnInferenceBase::~BiRnnInferenceBase ( )
virtualdefault

Member Function Documentation

◆ convert_dicts_and_embeddings()

void deeplima::eigen_impl::BiRnnInferenceBase::convert_dicts_and_embeddings ( const nets::BiRnnClassifierImpl &  src)
protectedvirtual

Definition at line 34 of file convert_from_torch.cpp.

◆ convert_from_torch()

virtual void deeplima::eigen_impl::BiRnnInferenceBase::convert_from_torch ( const std::string &  fn)
protectedpure virtual

Implemented in deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference, deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation, deeplima::tagging::eigen_impl::BiRnnEigenInferenceForTagging< AuxScalar >, and deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization.

◆ get_input_str_dicts()

const str_dicts_holder_t & deeplima::eigen_impl::BiRnnInferenceBase::get_input_str_dicts ( ) const
inline

Definition at line 47 of file birnn_inference_base.h.

◆ get_input_str_dicts_names()

const std::vector< std::string > & deeplima::eigen_impl::BiRnnInferenceBase::get_input_str_dicts_names ( ) const
inline

Definition at line 52 of file birnn_inference_base.h.

◆ get_input_uint_dicts()

const uint_dicts_holder_t & deeplima::eigen_impl::BiRnnInferenceBase::get_input_uint_dicts ( ) const
inline

Definition at line 37 of file birnn_inference_base.h.

◆ get_input_uint_dicts_names()

const std::vector< std::string > & deeplima::eigen_impl::BiRnnInferenceBase::get_input_uint_dicts_names ( ) const
inline

Definition at line 42 of file birnn_inference_base.h.

◆ get_output_str_dicts()

const std::vector< std::vector< std::string > > & deeplima::eigen_impl::BiRnnInferenceBase::get_output_str_dicts ( ) const
inline

Definition at line 57 of file birnn_inference_base.h.

◆ get_output_str_dicts_names()

const std::vector< std::string > & deeplima::eigen_impl::BiRnnInferenceBase::get_output_str_dicts_names ( ) const
inline

Definition at line 62 of file birnn_inference_base.h.

◆ init_new_worker()

virtual size_t deeplima::eigen_impl::BiRnnInferenceBase::init_new_worker ( size_t  input_len,
bool  precomputed_input = false 
)
inlinevirtual

Reimplemented in deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization.

Definition at line 67 of file birnn_inference_base.h.

◆ load()

virtual void deeplima::eigen_impl::BiRnnInferenceBase::load ( const std::string &  fn)
pure virtual

Implemented in deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference, deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation, deeplima::tagging::eigen_impl::BiRnnEigenInferenceForTagging< AuxScalar >, deeplima::RnnSeq2Seq, and deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization.

◆ precompute_inputs()

virtual void deeplima::eigen_impl::BiRnnInferenceBase::precompute_inputs ( const Eigen::MatrixXf &  inputs,
Eigen::MatrixXf &  outputs,
int64_t  input_size 
)
pure virtual

Implemented in deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation, deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference, deeplima::tagging::eigen_impl::BiRnnEigenInferenceForTagging< AuxScalar >, and deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization.

◆ predict()

virtual void deeplima::eigen_impl::BiRnnInferenceBase::predict ( size_t  worker_id,
const Eigen::MatrixXf &  inputs,
int64_t  input_begin,
int64_t  input_end,
int64_t  output_begin,
int64_t  output_end,
std::shared_ptr< StdMatrix< uint8_t > > &  output,
const std::vector< std::string > &  outputs_names 
)
pure virtual

Implemented in deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation, deeplima::tagging::eigen_impl::BiRnnEigenInferenceForTagging< AuxScalar >, deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference, and deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization.

Member Data Documentation

◆ m_input_str_dicts

str_dicts_holder_t deeplima::eigen_impl::BiRnnInferenceBase::m_input_str_dicts
protected

Definition at line 109 of file birnn_inference_base.h.

◆ m_input_str_dicts_names

std::vector<std::string> deeplima::eigen_impl::BiRnnInferenceBase::m_input_str_dicts_names
protected

Definition at line 110 of file birnn_inference_base.h.

◆ m_input_uint_dicts

uint_dicts_holder_t deeplima::eigen_impl::BiRnnInferenceBase::m_input_uint_dicts
protected

Definition at line 107 of file birnn_inference_base.h.

◆ m_input_uint_dicts_names

std::vector<std::string> deeplima::eigen_impl::BiRnnInferenceBase::m_input_uint_dicts_names
protected

Definition at line 108 of file birnn_inference_base.h.

◆ m_linear

std::vector<params_linear_t<Eigen::MatrixXf, Eigen::VectorXf> > deeplima::eigen_impl::BiRnnInferenceBase::m_linear
protected

Definition at line 123 of file birnn_inference_base.h.

◆ m_linear_idx

std::map<std::string, size_t> deeplima::eigen_impl::BiRnnInferenceBase::m_linear_idx
protected

Definition at line 124 of file birnn_inference_base.h.

◆ m_lstm

std::vector<params_bilstm_spec_t> deeplima::eigen_impl::BiRnnInferenceBase::m_lstm
protected

Definition at line 116 of file birnn_inference_base.h.

◆ m_lstm_idx

std::map<std::string, size_t> deeplima::eigen_impl::BiRnnInferenceBase::m_lstm_idx
protected

Definition at line 117 of file birnn_inference_base.h.

◆ m_multi_bilstm

std::vector<std::shared_ptr<params_multilayer_bilstm_spec_t> > deeplima::eigen_impl::BiRnnInferenceBase::m_multi_bilstm
protected

Definition at line 120 of file birnn_inference_base.h.

◆ m_multi_bilstm_idx

std::map<std::string, size_t> deeplima::eigen_impl::BiRnnInferenceBase::m_multi_bilstm_idx
protected

Definition at line 121 of file birnn_inference_base.h.

◆ m_ops

std::vector<std::shared_ptr<Op_Base> > deeplima::eigen_impl::BiRnnInferenceBase::m_ops
protected

Definition at line 102 of file birnn_inference_base.h.

◆ m_output_str_dicts

std::vector<std::vector<std::string> > deeplima::eigen_impl::BiRnnInferenceBase::m_output_str_dicts
protected

Definition at line 112 of file birnn_inference_base.h.

◆ m_output_str_dicts_names

std::vector<std::string> deeplima::eigen_impl::BiRnnInferenceBase::m_output_str_dicts_names
protected

Definition at line 113 of file birnn_inference_base.h.

◆ m_params

std::vector<std::shared_ptr<param_base_t> > deeplima::eigen_impl::BiRnnInferenceBase::m_params
protected

Definition at line 103 of file birnn_inference_base.h.

◆ m_wb

std::vector<std::vector<std::shared_ptr<Op_Base::workbench_t> > > deeplima::eigen_impl::BiRnnInferenceBase::m_wb
protected

Definition at line 105 of file birnn_inference_base.h.

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