pytorch-inference
pytorch::BatchNorm2d Member List

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

add_beta(const std::string &beta_filename="", const std::vector< int > &beta_dims={})pytorch::BatchNorm2dinline
add_gamma(const std::string &gamma_filename="", const std::vector< int > &gamma_dims={})pytorch::BatchNorm2dinline
add_running_mean(const std::string &running_mean_filename="", const std::vector< int > &running_mean_dims={})pytorch::BatchNorm2dinline
add_running_var(const std::string &running_var_filename="", const std::vector< int > &running_var_dims={})pytorch::BatchNorm2dinline
BatchNorm2d(const tensor &gamma, const tensor &beta, const tensor &running_mean, const tensor &running_var, const float &epsilon=1e-5)pytorch::BatchNorm2dinline
BatchNorm2d(const std::string &gamma_filename="", const std::vector< int > &gamma_dims={}, const std::string &beta_filename="", const std::vector< int > &beta_dims={}, const std::string &running_mean_filename="", const std::vector< int > &running_mean_dims={}, const std::string &running_var_filename="", const std::vector< int > &running_var_dims={}, const float &epsilon=1e-5, const std::string &python_home="../scripts")pytorch::BatchNorm2dinline
betapytorch::BatchNorm2dprivate
epsilonpytorch::BatchNorm2dprivate
forward(const std::vector< tensor > &input)pytorch::BatchNorm2dinlinevirtual
gammapytorch::BatchNorm2dprivate
operator()(const std::vector< tensor > &input)pytorch::BatchNorm2dinlinevirtual
running_meanpytorch::BatchNorm2dprivate
running_varpytorch::BatchNorm2dprivate
utilspytorch::BatchNorm2dprivate