LIMA
Libre Multilingual Analyzer — C++ API
Loading...
Searching...
No Matches
deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T > Member List

This is the complete list of members for deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >, including all inherited members.

create_workbench(uint32_t, const std::shared_ptr< param_base_t > params, bool precomputed_input=false) const overridedeeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlinevirtual
execute(std::shared_ptr< Op_Base::workbench_t > pwb, const EmbdUInt64Float &embd, const V &initial_state_h, const V &initial_state_c, const std::shared_ptr< param_base_t > params, const uint32_t start_code, const uint32_t stop_code, const size_t beam_size, std::vector< uint32_t > &output_seq, const size_t max_output_len)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlinevirtual
linear_and_decoding(const params_linear_t< M, V > &linear, const M &output, V &temp, const size_t beam_size, const size_t start_pos, std::vector< uint32_t > &top_classes, std::vector< uint32_t > indices, std::vector< float > &logprob)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
params_t typedefdeeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >
precompute_inputs(const std::shared_ptr< param_base_t > params, const M &inputs, M &outputs, int64_t first_column)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlinevirtual
step_fw(const size_t hidden_size, const size_t t, const V &s, V &c, M &output)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
step_fw(const size_t hidden_size, const size_t t, const V &s, M &c, const size_t beam_idx, M &output)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
step_with_decoding(const params_lstm_t< M, V > &layer, const params_linear_t< M, V > &linear, Eigen::Ref< const M > &input, M &output, V &s, V &temp, const size_t hidden_size, const size_t beam_size, const size_t start_pos, std::vector< uint32_t > &top_classes, std::vector< uint32_t > indices, std::vector< float > &logprob, V &c, V &h)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
step_with_decoding(const params_lstm_t< M, V > &layer, const params_linear_t< M, V > &linear, Eigen::Ref< const M > &input, M &output, V &s, V &temp, const size_t hidden_size, const size_t beam_size, const size_t start_pos, std::vector< uint32_t > &top_classes, std::vector< uint32_t > indices, std::vector< float > &logprob, M &states_c, M &states_h, const size_t beam_idx)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
supports_precomputing() constdeeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlinevirtual
update_c(const size_t hidden_size, const V &s, V &c)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
update_c(const size_t hidden_size, const V &s, M &c, const size_t beam_idx)deeplima::eigen_impl::Op_LSTM_Beam_Decoder< M, V, T >inlineprotected
~Op_Base()deeplima::eigen_impl::Op_Baseinlinevirtual