|
- // Tencent is pleased to support the open source community by making ncnn available.
- //
- // Copyright (C) 2022 THL A29 Limited, a Tencent company. All rights reserved.
- //
- // Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
- // in compliance with the License. You may obtain a copy of the License at
- //
- // https://opensource.org/licenses/BSD-3-Clause
- //
- // Unless required by applicable law or agreed to in writing, software distributed
- // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
- // CONDITIONS OF ANY KIND, either express or implied. See the License for the
- // specific language governing permissions and limitations under the License.
-
- #ifndef LAYER_DECONVOLUTIONDEPTHWISE1D_H
- #define LAYER_DECONVOLUTIONDEPTHWISE1D_H
-
- #include "layer.h"
-
- namespace ncnn {
-
- class DeconvolutionDepthWise1D : public Layer
- {
- public:
- DeconvolutionDepthWise1D();
-
- virtual int load_param(const ParamDict& pd);
-
- virtual int load_model(const ModelBin& mb);
-
- virtual int forward(const Mat& bottom_blob, Mat& top_blob, const Option& opt) const;
-
- virtual int forward(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;
-
- protected:
- void cut_padding(const Mat& top_blob_bordered, Mat& top_blob, const Option& opt) const;
-
- public:
- // param
- int num_output;
- int kernel_w;
- int dilation_w;
- int stride_w;
- int pad_left;
- int pad_right;
- int output_pad_right;
- int output_w;
- int bias_term;
-
- int weight_data_size;
- int group;
-
- int activation_type;
- Mat activation_params;
-
- int dynamic_weight;
-
- // model
- Mat weight_data;
- Mat bias_data;
- };
-
- } // namespace ncnn
-
- #endif // LAYER_DECONVOLUTIONDEPTHWISE1D_H
|