convert_from_torch.h File Reference

#include <torch/torch.h>
#include "static_graph/dict.h"
#include "deeplima/eigen_wrp/embd_dict.h"
#include "deeplima/eigen_wrp/deep_biaffine_attn_decoder.h"
#include "deeplima/eigen_wrp/deep_biaffine_attn_label_decoder.h"

Go to the source code of this file.

Namespaces
namespace	deeplima
namespace	deeplima::convert_from_torch

Functions
template<class V , class S >
void	deeplima::convert_from_torch::copy_vector (const torch::Tensor &src, V &dst)
template<class M , class S >
void	deeplima::convert_from_torch::copy_matrix (const torch::Tensor &src, M &dst)
template<class M , class V , class S = float>
void	deeplima::convert_from_torch::convert_module_from_torch (const torch::nn::LSTM &src, eigen_impl::params_bilstm_t< M, V > &dst)
template<class M , class V , class S = float>
void	deeplima::convert_from_torch::convert_module_from_torch (const torch::nn::Linear &src, eigen_impl::params_linear_t< M, V > &dst)
template<class M , class V , class S = float>
void	deeplima::convert_from_torch::convert_module_from_torch (const deeplima::nets::torch_modules::DeepBiaffineAttentionDecoder &src, deeplima::eigen_impl::params_deep_biaffine_attn_decoder_t< M, V > &dst)
template<class M , class V , class S = float>
void	deeplima::convert_from_torch::convert_module_from_torch (const deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoder &src, deeplima::eigen_impl::params_deep_biaffine_attn_label_decoder_t< M, V > &dst)
template<class M , class S = float>
void	deeplima::convert_from_torch::convert_module_from_torch (const torch::nn::Embedding &src, std::shared_ptr< DictBase > sp_dict, EmbdDict< uint64_t, M > &dst)
template<class M , class S = float>
void	deeplima::convert_from_torch::convert_module_from_torch (const torch::nn::Embedding &src, std::shared_ptr< DictBase > sp_dict, EmbdDict< std::string, M > &dst)