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 */