|
| virtual std::shared_ptr< Op_Base::workbench_t > | create_workbench (uint32_t input_size, const std::shared_ptr< param_base_t > params, bool precomputed_input=false) const override |
| |
| virtual bool | supports_precomputing () const |
| |
| virtual void | precompute_inputs (const std::shared_ptr< param_base_t > params, const M &inputs, M &outputs, int64_t first_column) |
| |
| virtual size_t | execute (std::shared_ptr< Op_Base::workbench_t > pwb, const M &input_matrix, const std::shared_ptr< param_base_t > params, size_t input_begin, size_t input_end, Vector &fw_h, Vector &fw_c, Vector &bw_h, Vector &bw_c) |
| |
| virtual size_t | execute (std::shared_ptr< Op_Base::workbench_t > pwb, const M &input_matrix, const std::shared_ptr< param_base_t > params, const size_t input_begin, const size_t input_end) |
| |
| virtual | ~Op_Base () |
| |
|
| void | forward_pass (size_t hidden_size, const params_lstm_t< M, V > &fw, M &input, V &s, V &g_u, V &g_o, V &g_if, V &c, M &output, const V &initial_h, int begin, int end) |
| |
| void | backward_pass (size_t hidden_size, const params_lstm_t< M, V > &bw, M &input, V &s, V &g_u, V &g_o, V &g_if, V &c, M &output, const V &initial_h, int begin, int end) |
| |
| void | step_fw (size_t hidden_size, size_t t, const V &s, V &g_u, V &g_o, V &g_if, V &c, M &output) |
| |
| void | step_bw (size_t hidden_size, size_t t, const V &s, V &g_u, V &g_o, V &g_if, V &c, M &output) |
| |
template<class M, class V, class T>
class deeplima::eigen_impl::Op_BiLSTM< M, V, T >
Definition at line 100 of file bilstm.h.