diff options
Diffstat (limited to 'compute/ncnn/src/srcn/conv_sparse.h')
-rw-r--r-- | compute/ncnn/src/srcn/conv_sparse.h | 79 |
1 files changed, 0 insertions, 79 deletions
diff --git a/compute/ncnn/src/srcn/conv_sparse.h b/compute/ncnn/src/srcn/conv_sparse.h deleted file mode 100644 index 7ac358fd8..000000000 --- a/compute/ncnn/src/srcn/conv_sparse.h +++ /dev/null @@ -1,79 +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_SPARSE_H__ -#define __NNFW_SRCN_CONV_SPARSE_H__ - -#include "ncnn/srcn/conv_type.h" -#include "common.h" - -namespace nnfw -{ -namespace srcn -{ - -#define BCH 128 - -typedef struct -{ - short m; - short k; - float data; -} weight_data_t; - -typedef struct -{ - int mxk; - weight_data_t *wdata; -} sparse_weight_t; - -class conv_sparse -{ -public: - conv_sparse(const convMat_t &in_mat, convMat_t &out_mat, const convParams_t &in_param, - const sparse_weight_t *weights, int num_threads, convType_t conv_type); - ~conv_sparse(); - - void run(); - -private: - void param_init(); - void compute_singlethread(); - void compute_multithreads(); - - const convMat_t in_mat_; - convMat_t out_mat_; - const convParams_t in_param_; - const sparse_weight_t *weights_; - int num_threads_; - convType_t conv_type_; - - uint32_t n_; - uint32_t bn_; - int rn_; - int nn_; - - int bch_; - int rch_; - int nch_; - - int need_im2col_; -}; - -} // namespace srcn -} // namespace nnfw - -#endif // __NNFW_SRCN_CONV_SPARSE_H__ |