Blame view

3rdparty/opencv-4.5.4/modules/dnn/src/halide_scheduler.hpp 937 Bytes
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
  // This file is part of OpenCV project.
  // It is subject to the license terms in the LICENSE file found in the top-level directory
  // of this distribution and at http://opencv.org/license.html.
  //
  // Copyright (C) 2017, Intel Corporation, all rights reserved.
  // Third party copyrights are property of their respective owners.
  
  #ifndef __OPENCV_DNN_HALIDE_SCHEDULER_HPP__
  #define __OPENCV_DNN_HALIDE_SCHEDULER_HPP__
  
  #include <opencv2/dnn.hpp>
  
  namespace cv
  {
  namespace dnn
  {
  
  class HalideScheduler
  {
  public:
      HalideScheduler(const std::string& configFile);
  
      ~HalideScheduler();
  
      // Returns true if pipeline found in scheduling file.
      // If more than one function, returns true if the top function scheduled.
      // Other functions are optional to scheduling.
      bool process(Ptr<BackendNode>& node);
  
  private:
      FileStorage fs;
  };
  
  }  // namespace dnn
  }  // namespace cv
  
  #endif  // __OPENCV_DNN_HALIDE_SCHEDULER_HPP__