Caffe2 - C++ API
A deep learning, cross platform ML framework
mpscnn_graph_mask.h
1 
18 #pragma once
19 #include "caffe2/core/net.h"
20 #include "mpscnn.h"
21 
22 namespace caffe2 {
23 // We currently only try to convert a fixed set of operators that handle a subset of a full
24 // CNN. We also only run when MPSCNN is available, provides a speedup.
25 // On failure, returns false. On success, returns true, and sets the MPSCNN net in the output
26 // parameter.
27 // The rewrite function now supports insertion of copies in intermediate ops.
28 bool tryConvertToMPSCNNIntermediateCopies(const NetDef& initNet,
29  const NetDef& predictNet,
30  NetDef* mpscnnPredictNet);
31 NetDef setSpecialArgs(const NetDef& def);
32 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.