Caffe2 - C++ API
A deep learning, cross platform ML framework
optimize_ideep.h
1 #pragma once
2 
3 #include "caffe2/core/common.h"
4 #include "caffe2/core/workspace.h"
5 #include "caffe2/proto/caffe2_pb.h"
6 #include "nomnigraph/Representations/NeuralNet.h"
7 
8 namespace caffe2 {
9 namespace opt {
10 
11 CAFFE2_API void OptimizeForIdeep(
12  nom::repr::NNModule* nn,
13  caffe2::Workspace* ws,
14  bool training_mode = false);
15 }
16 } // namespace caffe2
caffe2::Workspace
Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
Definition: workspace.h:47
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13
Generated on Thu Mar 21 2019 13:06:20 for Caffe2 - C++ API by   doxygen 1.8.11