1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements.  See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership.  The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * License); you may not use this file except in compliance
8  * with the License.  You may obtain a copy of the License at
9  *
10  *   http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * AS IS BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied.  See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  */
19 
20 /*
21  * Copyright (c) 2020, OPEN AI LAB
22  * Author: qtang@openailab.com
23  */
24 
25 #ifndef TENGINE_GRAPH_CONVOLUTION_HPP
26 #define TENGINE_GRAPH_CONVOLUTION_HPP
27 
28 #define FLOAT_TO_REALSIZE (4)
29 #ifdef HAVE_TENGINE
30 
31 #include "tengine_c_api.h"
32 
33 namespace cv
34 {
35 namespace dnn
36 {
37 teng_graph_t  tengine_init(const char* name , float* input_, int inch, int group, int in_h, int in_w,
38                         float *output_, int out_b, int outch, int out_h, int out_w,
39                         float *kernel_,int kernel_s , int kernel_h, int kernel_w,
40                         float *teg_bias, int stride_h,int stride_w,
41                         int pad_h, int pad_w,  int dilation_h, int dilation_w,
42                         size_t wstep, const std::string padMode , teng_graph_t& graph, int nstripes) ;
43 
44 bool tengine_forward(teng_graph_t& graph) ;
45 bool tengine_release(teng_graph_t& graph) ;
46 }
47 }
48 #endif
49 #endif /* TENGINE_GRAPH_CONVOLUTION_HPP */