| 13 | } |
| 14 | |
| 15 | LinearBnReluImpl::LinearBnReluImpl(int in_features, int out_features){ |
| 16 | ln = register_module("ln", torch::nn::Linear(torch::nn::LinearOptions(in_features, out_features))); |
| 17 | bn = register_module("bn", torch::nn::BatchNorm1d(out_features)); |
| 18 | } |
| 19 | |
| 20 | torch::Tensor LinearBnReluImpl::forward(torch::Tensor x){ |
| 21 | x = torch::relu(ln->forward(x)); |
nothing calls this directly
no outgoing calls
no test coverage detected