| 17 | } |
| 18 | |
| 19 | void CaffeExtraManager::insert(const std::string& name, std::shared_ptr<Transform> transform) { |
| 20 | OpCount::get()->insertOp("CAFFE", name); |
| 21 | mTransform.insert(std::make_pair(name, transform)); |
| 22 | } |
| 23 | std::shared_ptr<CaffeExtraManager::Transform> CaffeExtraManager::find(const std::string& name) const { |
| 24 | auto iter = mTransform.find(name); |
| 25 | if (iter == mTransform.end()) { |
no test coverage detected