Caffe2 - C++ API
A deep learning, cross platform ML framework
conv_op_shared.h
1 #ifndef CAFFE2_OPERATORS_CONV_OP_SHARED_H_
2 #define CAFFE2_OPERATORS_CONV_OP_SHARED_H_
3 
4 #include "caffe2/core/context.h"
5 #include "caffe2/core/tensor.h"
6 #include "caffe2/core/workspace.h"
7 
8 namespace caffe2 {
9 
14 template <typename Context>
15 void createSharedBuffer(Workspace* ws);
16 
21 template <typename Context>
22 void runWithSharedBuffer(
23  Workspace* ws,
24  std::function<void(Tensor<Context>* buffer)> f);
25 } // namespace caffe2
26 
27 #endif // CAFFE2_OPERATORS_CONV_OP_SHARED_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
caffe2::createSharedBuffer
void createSharedBuffer(Workspace *ws)
Creates a mutex and shared buffer in the workspace.
Generated on Thu Apr 19 2018 13:03:55 for Caffe2 - C++ API by   doxygen 1.8.11