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
arm-compute
core
operator.h
1
#ifndef CAFFE2_OPENGL_OPERATOR_H_
2
#define CAFFE2_OPENGL_OPERATOR_H_
3
4
#include "caffe2/core/operator.h"
5
#include "caffe2/core/registry.h"
6
7
namespace
caffe2
{
8
9
CAFFE_DECLARE_REGISTRY(GLOperatorRegistry, OperatorBase,
const
OperatorDef &,
10
Workspace *);
11
#define REGISTER_GL_OPERATOR_CREATOR(key, ...) \
12
CAFFE_REGISTER_CREATOR(GLOperatorRegistry, key, __VA_ARGS__)
13
#define REGISTER_GL_OPERATOR(name, ...) \
14
extern void CAFFE2_PLEASE_ADD_OPERATOR_SCHEMA_FOR_##name(); \
15
static void CAFFE2_UNUSED CAFFE_ANONYMOUS_VARIABLE_GL##name() { \
16
CAFFE2_PLEASE_ADD_OPERATOR_SCHEMA_FOR_##name(); \
17
} \
18
CAFFE_REGISTER_CLASS(GLOperatorRegistry, name, __VA_ARGS__)
19
#define REGISTER_GL_OPERATOR_STR(str_name, ...) \
20
CAFFE_REGISTER_TYPED_CLASS(GLOperatorRegistry, str_name, __VA_ARGS__)
21
22
#define REGISTER_GL_OPERATOR_WITH_ENGINE(name, engine, ...) \
23
CAFFE_REGISTER_CLASS(GLOperatorRegistry, name##_ENGINE_##engine, __VA_ARGS__)
24
25
}
// namespace caffe2
26
27
#endif // CAFFE2_OPENGL_OPERATOR_H_
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