Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
mobile
contrib
opengl
core
rewrite_net.h
1
2
#pragma once
3
#include "GLPredictor.h"
4
#include "caffe2/core/predictor.h"
5
6
namespace
caffe2
{
7
bool
tryConvertToOpenGL(
const
NetDef& initNet,
8
const
NetDef& predictNet,
9
NetDef* glPredictNet,
10
bool
useTextureInput =
false
,
11
bool
useTiling =
false
,
12
bool
runFusion =
true
);
13
14
// Exposed for testing
15
NetDef rewritePredictNetForOpenGL(
const
NetDef& predictNet,
16
bool
useTextureInput =
false
,
17
bool
useTiling =
false
,
18
bool
runFusion =
true
);
19
void
dumpDefForOpenGL(
const
NetDef& net);
20
}
// namespace caffe2
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
convert_encoded_to_raw_leveldb.cc:47
Generated on Thu Apr 19 2018 13:03:50 for Caffe2 - C++ API by
1.8.11