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

#include <deeplima/eigen_wrp/graph_dp_eigen_inference_impl.h>

Inheritance diagram for deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference:
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 ~BiRnnAndDeepBiaffineAttentionEigenInference ()=default
 
virtual void load (const std::string &fn)
 
virtual size_t get_precomputed_dim () const
 
virtual void precompute_inputs (const Eigen::MatrixXf &inputs, Eigen::MatrixXf &outputs, int64_t input_size)
 
virtual void predict (size_t, const Eigen::MatrixXf &, int64_t, int64_t, int64_t, int64_t, std::shared_ptr< StdMatrix< uint8_t > > &, const std::vector< std::string > &)
 
virtual void predict (size_t worker_id, const Eigen::MatrixXf &inputs, int64_t input_begin, int64_t, int64_t, int64_t, std::shared_ptr< StdMatrix< uint32_t > > &output, const std::vector< size_t > &lengths, const std::vector< std::string > &)
 
const std::string & get_embd_fn (size_t idx) const
 
const std::vector< std::string > & get_rel_class_names () const
 
bool has_label_decoder () const
 
- 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< std::string > m_embd_fn
 
std::vector< std::shared_ptr< deeplima::eigen_impl::params_deep_biaffine_attn_decoder_t< Eigen::MatrixXf, Eigen::VectorXf > > > m_deep_biaffine_attn_decoder
 
std::map< std::string, size_t > m_deep_biaffine_attn_decoder_idx
 
std::vector< std::shared_ptr< deeplima::eigen_impl::params_deep_biaffine_attn_label_decoder_t< Eigen::MatrixXf, Eigen::VectorXf > > > m_deep_biaffine_attn_label_decoder
 
std::vector< std::string > m_rel_class_names
 
- 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 34 of file graph_dp_eigen_inference_impl.h.

Member Typedef Documentation

◆ dicts_holder_t

◆ Matrix

◆ Parent

◆ Scalar

◆ tensor_t

◆ Vector

Constructor & Destructor Documentation

◆ ~BiRnnAndDeepBiaffineAttentionEigenInference()

virtual deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::~BiRnnAndDeepBiaffineAttentionEigenInference ( )
virtualdefault

Member Function Documentation

◆ convert_from_torch()

void deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::convert_from_torch ( const std::string &  fn)
protectedvirtual

Implements deeplima::eigen_impl::BiRnnInferenceBase.

Definition at line 27 of file graph_dp_model.cpp.

◆ get_embd_fn()

const std::string & deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::get_embd_fn ( size_t  idx) const
inline

Definition at line 149 of file graph_dp_eigen_inference_impl.h.

◆ get_precomputed_dim()

virtual size_t deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::get_precomputed_dim ( ) const
inlinevirtual

Definition at line 52 of file graph_dp_eigen_inference_impl.h.

◆ get_rel_class_names()

const std::vector< std::string > & deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::get_rel_class_names ( ) const
inline

Definition at line 155 of file graph_dp_eigen_inference_impl.h.

◆ has_label_decoder()

bool deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::has_label_decoder ( ) const
inline

Definition at line 160 of file graph_dp_eigen_inference_impl.h.

◆ load()

virtual void deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::load ( const std::string &  fn)
inlinevirtual

◆ precompute_inputs()

virtual void deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::precompute_inputs ( const Eigen::MatrixXf &  inputs,
Eigen::MatrixXf &  outputs,
int64_t  input_size 
)
inlinevirtual

◆ predict() [1/2]

virtual void deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::predict ( size_t  worker_id,
const Eigen::MatrixXf &  inputs,
int64_t  input_begin,
int64_t  ,
int64_t  ,
int64_t  ,
std::shared_ptr< StdMatrix< uint32_t > > &  output,
const std::vector< size_t > &  lengths,
const std::vector< std::string > &   
)
inlinevirtual

Definition at line 86 of file graph_dp_eigen_inference_impl.h.

◆ predict() [2/2]

virtual void deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::predict ( size_t  ,
const Eigen::MatrixXf &  ,
int64_t  ,
int64_t  ,
int64_t  ,
int64_t  ,
std::shared_ptr< StdMatrix< uint8_t > > &  ,
const std::vector< std::string > &   
)
inlinevirtual

Member Data Documentation

◆ m_deep_biaffine_attn_decoder

std::vector<std::shared_ptr<deeplima::eigen_impl::params_deep_biaffine_attn_decoder_t<Eigen::MatrixXf, Eigen::VectorXf> > > deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::m_deep_biaffine_attn_decoder
protected

Definition at line 168 of file graph_dp_eigen_inference_impl.h.

◆ m_deep_biaffine_attn_decoder_idx

std::map<std::string, size_t> deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::m_deep_biaffine_attn_decoder_idx
protected

Definition at line 169 of file graph_dp_eigen_inference_impl.h.

◆ m_deep_biaffine_attn_label_decoder

std::vector<std::shared_ptr<deeplima::eigen_impl::params_deep_biaffine_attn_label_decoder_t<Eigen::MatrixXf, Eigen::VectorXf> > > deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::m_deep_biaffine_attn_label_decoder
protected

Definition at line 171 of file graph_dp_eigen_inference_impl.h.

◆ m_embd_fn

std::vector<std::string> deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::m_embd_fn
protected

Definition at line 166 of file graph_dp_eigen_inference_impl.h.

◆ m_rel_class_names

std::vector<std::string> deeplima::graph_dp::eigen_impl::BiRnnAndDeepBiaffineAttentionEigenInference::m_rel_class_names
protected

Definition at line 172 of file graph_dp_eigen_inference_impl.h.


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