| 117 | } |
| 118 | |
| 119 | bool TFSerializer::LoadModel(const std::vector<std::string>& file_list, StaticGraph* graph) |
| 120 | { |
| 121 | tensorflow::GraphDef tf_net; |
| 122 | |
| 123 | if( //! LoadTextFile(file_list[0].c_str(), tf_net) && |
| 124 | !LoadBinaryFile(file_list[0].c_str(), tf_net)) |
| 125 | return false; |
| 126 | |
| 127 | SetGraphSource(graph, file_list[0]); |
| 128 | SetGraphSourceFormat(graph, "tensorflow"); |
| 129 | SetGraphConstTensorFile(graph, file_list[0]); |
| 130 | SetGraphLayout(graph, TENGINE_LAYOUT_NHWC); |
| 131 | SetModelLayout(graph, TENGINE_LAYOUT_NHWC); |
| 132 | SetModelFormat(graph, MODEL_FORMAT_TENSORFLOW); |
| 133 | |
| 134 | return LoadGraph(tf_net, graph); |
| 135 | } |
| 136 | |
| 137 | bool TFSerializer::LoadTextFile(const char* fname, tensorflow::GraphDef& tf_net) |
| 138 | { |