Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::memonger::ComputeBlobRecyclingForDag Class Reference

Public Member Functions

 ComputeBlobRecyclingForDag (const int size)
 
NetDef OptimizeNet (const NetDef &net, const std::vector< string > &heads, const std::vector< int > &op_indices, const std::unordered_set< string > &shareable_blob_names, const string &namescope, const std::unordered_set< string > &dont_share_blob_names, const std::unordered_map< string, vector< int >> &blob_shapes)
 

Detailed Description

Definition at line 126 of file memonger.cc.


The documentation for this class was generated from the following file: