| 9 | #define slots Q_SLOTS |
| 10 | |
| 11 | inline torch::nn::Conv2dOptions conv_options(int64_t in_planes, int64_t out_planes, int64_t kerner_size, |
| 12 | int64_t stride = 1, int64_t padding = 0, bool with_bias = false) { |
| 13 | torch::nn::Conv2dOptions conv_options = torch::nn::Conv2dOptions(in_planes, out_planes, kerner_size); |
| 14 | conv_options.stride(stride); |
| 15 | conv_options.padding(padding); |
| 16 | conv_options.bias(with_bias); |
| 17 | return conv_options; |
| 18 | } |
| 19 | |
| 20 | class ConvReluBnImpl : public torch::nn::Module { |
| 21 | public: |
no outgoing calls
no test coverage detected