|
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 |