Caffe2 - C++ API
A deep learning, cross platform ML framework
opengl_test.h
1 
18 #include "caffe2/proto/caffe2.pb.h"
19 
20 namespace caffe2 {
21 void testOpenGL();
22 void compareModelsForOpenGL(std::string name,
23  const NetDef& initNet,
24  NetDef predictNet,
25  int width,
26  int height,
27  int channel,
28  std::string input_type,
29  std::string input_order);
30 
31 void compareBatchedToTiledModels(std::string name,
32  const NetDef& initNet,
33  NetDef predictNet,
34  int width,
35  int height,
36  int channel,
37  std::string input_type,
38  std::string input_order);
39 
40 int runModelBenchmarks(caffe2::NetDef& init_net,
41  caffe2::NetDef& predict_net,
42  int warm_up_runs,
43  int main_runs,
44  int channel,
45  int height,
46  int width,
47  std::string input_type,
48  std::string input_order,
49  std::string engine,
50  bool run_individual = false,
51  bool use_texture_input = false,
52  bool use_tiling = false,
53  bool run_fusion = true);
54 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.