pytorch-inference
|
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::BatchNorm2d | inline |
add_gamma(const std::string &gamma_filename="", const std::vector< int > &gamma_dims={}) | pytorch::BatchNorm2d | inline |
add_running_mean(const std::string &running_mean_filename="", const std::vector< int > &running_mean_dims={}) | pytorch::BatchNorm2d | inline |
add_running_var(const std::string &running_var_filename="", const std::vector< int > &running_var_dims={}) | pytorch::BatchNorm2d | inline |
BatchNorm2d(const tensor &gamma, const tensor &beta, const tensor &running_mean, const tensor &running_var, const float &epsilon=1e-5) | pytorch::BatchNorm2d | inline |
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::BatchNorm2d | inline |
beta | pytorch::BatchNorm2d | private |
epsilon | pytorch::BatchNorm2d | private |
forward(const std::vector< tensor > &input) | pytorch::BatchNorm2d | inlinevirtual |
gamma | pytorch::BatchNorm2d | private |
operator()(const std::vector< tensor > &input) | pytorch::BatchNorm2d | inlinevirtual |
running_mean | pytorch::BatchNorm2d | private |
running_var | pytorch::BatchNorm2d | private |
utils | pytorch::BatchNorm2d | private |