LIMA
Libre Multilingual Analyzer — C++ API
Loading...
Searching...
No Matches
deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation Class Reference

#include <deeplima/eigen_wrp/segmentation_eigen_inference_impl.h>

Inheritance diagram for deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation:
deeplima::eigen_impl::BiRnnInferenceBase

Public Types

typedef Eigen::MatrixXf Matrix
 
typedef Eigen::VectorXf Vector
 
typedef float Scalar
 
typedef Eigen::MatrixXf tensor_t
 
typedef EmbdUInt64FloatHolder dicts_holder_t
 
typedef deeplima::eigen_impl::BiRnnInferenceBase Parent
 
- Public Types inherited from deeplima::eigen_impl::BiRnnInferenceBase
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 void load (const std::string &fn)
 
const std::vector< impl::ngram_descr_t > & get_ngram_descr () const
 
virtual void precompute_inputs (const Eigen::MatrixXf &, Eigen::MatrixXf &, int64_t)
 
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 > &)
 
- Public Member Functions inherited from deeplima::eigen_impl::BiRnnInferenceBase
virtual ~BiRnnInferenceBase ()=default
 
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)
 

Protected Member Functions

virtual void convert_from_torch (const std::string &fn)
 
- Protected Member Functions inherited from deeplima::eigen_impl::BiRnnInferenceBase
virtual void convert_dicts_and_embeddings (const nets::BiRnnClassifierImpl &src)
 

Protected Attributes

std::vector< impl::ngram_descr_t > m_ngram_gescr
 
- Protected Attributes inherited from deeplima::eigen_impl::BiRnnInferenceBase
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
 

Additional Inherited Members

- Protected Types inherited from deeplima::eigen_impl::BiRnnInferenceBase
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
 

Detailed Description

Definition at line 35 of file segmentation_eigen_inference_impl.h.

Member Typedef Documentation

◆ dicts_holder_t

◆ Matrix

◆ Parent

◆ Scalar

◆ tensor_t

◆ Vector

Member Function Documentation

◆ convert_from_torch()

void deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation::convert_from_torch ( const std::string &  fn)
protectedvirtual

Implements deeplima::eigen_impl::BiRnnInferenceBase.

Definition at line 124 of file convert_from_torch.cpp.

◆ get_ngram_descr()

const std::vector< impl::ngram_descr_t > & deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation::get_ngram_descr ( ) const
inline

Definition at line 50 of file segmentation_eigen_inference_impl.h.

◆ load()

virtual void deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation::load ( const std::string &  fn)
inlinevirtual

◆ precompute_inputs()

virtual void deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation::precompute_inputs ( const Eigen::MatrixXf &  ,
Eigen::MatrixXf &  ,
int64_t   
)
inlinevirtual

◆ predict()

virtual void deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation::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 > &   
)
inlinevirtual

Member Data Documentation

◆ m_ngram_gescr

std::vector<impl::ngram_descr_t> deeplima::segmentation::eigen_impl::BiRnnEigenInferenceForSegmentation::m_ngram_gescr
protected

Definition at line 83 of file segmentation_eigen_inference_impl.h.


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