diff options
Diffstat (limited to 'compute/ncnn/src/srcn/conv_winograd.h')
-rw-r--r-- | compute/ncnn/src/srcn/conv_winograd.h | 72 |
1 files changed, 0 insertions, 72 deletions
diff --git a/compute/ncnn/src/srcn/conv_winograd.h b/compute/ncnn/src/srcn/conv_winograd.h deleted file mode 100644 index 76c2601f2..000000000 --- a/compute/ncnn/src/srcn/conv_winograd.h +++ /dev/null @@ -1,72 +0,0 @@ -/* - * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * 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 __NNFW_SRCN_CONV_WINOGRAD_H__ -#define __NNFW_SRCN_CONV_WINOGRAD_H__ - -#include "ncnn/srcn/conv_type.h" -#include "winograd.h" -#include "sgemm_singlethread.h" - -namespace nnfw -{ -namespace srcn -{ - -class conv_winograd -{ -public: - conv_winograd(const convMat_t &in_mat, convMat_t &out_mat, const convParams_t &in_param, - convType_t conv_type, const float *winograd_weight, int num_threads, int inc_stride, - int outc_stride, int c_stride); - ~conv_winograd(); - - void run(); - -private: - void param_init(); - void compute_sgemm(sgemmType_t major_type, sgemmTrans_t ltrans, sgemmTrans_t rtrans, const int m, - const int n, const int k, const float *lhs_data, const float *rhs_data, - float *res_data); - void winograd_input_im2col(float *col_buff); - void winograd_output_col2im(const float *col_buff); - void compute_winograd(); - - const convMat_t in_mat_; - convMat_t out_mat_; - const convParams_t in_param_; - convType_t conv_type_; - const float *winograd_weight_; - const int num_threads_; - - int tile_w_in_; - int tile_h_in_; - int tile_w_out_; - int tile_h_out_; - int ntiles_w_; - int ntiles_h_; - - int inc_stride_; - int outc_stride_; - int c_stride_; - - int error_; -}; - -} // namespace srcn -} // namespace nnfw - -#endif // __NNFW_SRCN_CONV_WINOGRAD_H__ |