pytorch-inference
pytorch::Linear Member List

This is the complete list of members for pytorch::Linear, including all inherited members.

add_bias(const std::string &bias_filename, const std::vector< int > &bias_dims)pytorch::Linearinline
add_weights(const std::string &weights_filename, const std::vector< int > &weights_dims)pytorch::Linearinline
biaspytorch::Linearprivate
forward(const std::vector< tensor > &input)pytorch::Linearinlinevirtual
has_biaspytorch::Linearprivate
Linear(const tensor &weights, const tensor &bias)pytorch::Linearinline
Linear(const std::string &weights_filename="", const std::vector< int > &weights_dims={}, const std::string &bias_filename="", const std::vector< int > &bias_dims={}, const std::string &python_home="../scripts")pytorch::Linearinline
Linear(const Linear &other)pytorch::Linearinline
operator()(const std::vector< tensor > &input)pytorch::Linearinlinevirtual
utilspytorch::Linearprivate
weightspytorch::Linearprivate
~Linear()pytorch::Linearinlinevirtual