| 130 | } |
| 131 | |
| 132 | std::vector<Weights> CaffeWeightFactory::getAllWeights(const std::string& layerName) |
| 133 | { |
| 134 | std::vector<Weights> v; |
| 135 | for (int i = 0;; i++) |
| 136 | { |
| 137 | auto b = getBlob(layerName, i); |
| 138 | if (b == nullptr) |
| 139 | { |
| 140 | break; |
| 141 | } |
| 142 | auto weights = getWeights(*b, layerName); |
| 143 | convert(weights, DataType::kFLOAT); |
| 144 | v.push_back(weights); |
| 145 | } |
| 146 | return v; |
| 147 | } |
| 148 | |
| 149 | Weights CaffeWeightFactory::operator()(const std::string& layerName, WeightType weightType) |
| 150 | { |
no outgoing calls
no test coverage detected