diff options
author | Jeff Donahue <jeff.donahue@gmail.com> | 2015-01-19 14:49:39 -0800 |
---|---|---|
committer | Jeff Donahue <jeff.donahue@gmail.com> | 2015-02-13 17:28:10 -0800 |
commit | e2358512dd36548c0080b0f381477418134b480f (patch) | |
tree | 4eb6fc892b83c323b1da343ec4a5431f165931ea /include | |
parent | c15d1846969ab8338561489194d787676733caa7 (diff) | |
download | caffeonacl-e2358512dd36548c0080b0f381477418134b480f.tar.gz caffeonacl-e2358512dd36548c0080b0f381477418134b480f.tar.bz2 caffeonacl-e2358512dd36548c0080b0f381477418134b480f.zip |
add Net::param_owners accessor for param sharing info
Diffstat (limited to 'include')
-rw-r--r-- | include/caffe/net.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/include/caffe/net.hpp b/include/caffe/net.hpp index 2709d87e..23da5dbb 100644 --- a/include/caffe/net.hpp +++ b/include/caffe/net.hpp @@ -147,6 +147,7 @@ class Net { const map<string, int>& param_names_index() const { return param_names_index_; } + inline const vector<int>& param_owners() const { return param_owners_; } /// @brief Input and output blob numbers inline int num_inputs() const { return net_input_blobs_.size(); } inline int num_outputs() const { return net_output_blobs_.size(); } |