| 213 | } |
| 214 | |
| 215 | bool CaffeSingle::LoadModel(const std::vector<std::string>& file_list, StaticGraph* graph) |
| 216 | { |
| 217 | te_caffe::NetParameter caffe_net; |
| 218 | |
| 219 | if(file_list.size() != GetFileNum()) |
| 220 | return false; |
| 221 | |
| 222 | if(!LoadBinaryFile(file_list[0].c_str(), caffe_net)) |
| 223 | return false; |
| 224 | |
| 225 | SetGraphSource(graph, file_list[0]); |
| 226 | SetGraphSourceFormat(graph, "caffe"); |
| 227 | SetGraphConstTensorFile(graph, file_list[0]); |
| 228 | SetGraphLayout(graph, TENGINE_LAYOUT_NCHW); |
| 229 | SetModelLayout(graph, TENGINE_LAYOUT_NCHW); |
| 230 | SetModelFormat(graph, MODEL_FORMAT_CAFFE); |
| 231 | |
| 232 | return LoadGraph(caffe_net, graph); |
| 233 | } |
| 234 | |
| 235 | bool CaffeSingle::LoadNode(StaticGraph* graph, StaticNode* node, const te_caffe::LayerParameter& layer_param, |
| 236 | name_map_t& tensor_name_map) |
nothing calls this directly
no test coverage detected