Caffe2 - C++ API
A deep learning, cross platform ML framework
rewrite_net.h
1 
18 #pragma once
19 #include "GLPredictor.h"
20 #include "caffe2/core/predictor.h"
21 
22 namespace caffe2 {
23 bool tryConvertToOpenGL(const NetDef& initNet,
24  const NetDef& predictNet,
25  NetDef* glPredictNet,
26  bool useTextureInput = false,
27  bool useTiling = false,
28  bool runFusion = true);
29 
30 // Exposed for testing
31 NetDef rewritePredictNetForOpenGL(const NetDef& predictNet,
32  bool useTextureInput = false,
33  bool useTiling = false,
34  bool runFusion = true);
35 void dumpDefForOpenGL(const NetDef& net);
36 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.