Caffe2 - C++ API
A deep learning, cross platform ML framework
conv_op_shared.h
1 
17 #ifndef CAFFE2_OPERATORS_CONV_OP_SHARED_H_
18 #define CAFFE2_OPERATORS_CONV_OP_SHARED_H_
19 
20 #include "caffe2/core/context.h"
21 #include "caffe2/core/tensor.h"
22 #include "caffe2/core/workspace.h"
23 
24 namespace caffe2 {
25 
30 template <typename Context>
31 void createSharedBuffer(Workspace* ws);
32 
37 template <typename Context>
38 void runWithSharedBuffer(
39  Workspace* ws,
40  std::function<void(Tensor<Context>* buffer)> f);
41 } // namespace caffe2
42 
43 #endif // CAFFE2_OPERATORS_CONV_OP_SHARED_H_
Copyright (c) 2016-present, Facebook, Inc.
Copyright (c) 2016-present, Facebook, Inc.
void createSharedBuffer(Workspace *ws)
Creates a mutex and shared buffer in the workspace.