reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-19 611d12371729d0e9f4a36b74d08f5c2deb4b71c8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#pragma once
 
#include "caffe2/core/net.h"
#include "caffe2/core/workspace.h"
#include "caffe2/predictor/InferenceGraph.h"
 
namespace caffe2 {
 
 
// Make all operators of a given type inplace when possible
void InPlaceOps(const InferenceGraph& graph_, const std::string& op_type);
 
void RemoveOpsByType(const InferenceGraph& graph_, const std::string& op_type);
 
} // namespace caffe2