We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ace506f commit 254de9eCopy full SHA for 254de9e
caffe2/core/net.h
@@ -17,7 +17,6 @@
17
#include "caffe2/core/observer.h"
18
#include "caffe2/core/operator_schema.h"
19
#include "caffe2/core/tensor.h"
20
-#include "caffe2/core/workspace.h"
21
#include "caffe2/proto/caffe2_pb.h"
22
#include "caffe2/utils/simple_queue.h"
23
caffe2/predictor/predictor.cc
@@ -4,6 +4,7 @@
4
5
namespace caffe2 {
6
7
+class Workspace;
8
namespace {
9
10
void enforceIsTensor(Workspace* ws, const std::string& name) {
caffe2/predictor/predictor_config.h
@@ -1,7 +1,8 @@
1
#pragma once
2
#include <memory>
3
-#include "caffe2/core/net.h"
+
+#include "caffe2/core/workspace.h"
#include "caffe2/proto/metanet.pb.h"
#include "caffe2/proto/predictor_consts.pb.h"
0 commit comments