Skip to content
1 change: 0 additions & 1 deletion app/Graph/acc_check.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
#include <sstream>
#include <unordered_map>

#include "build.cpp"
#include "build.hpp"

namespace fs = std::filesystem;
Expand Down
7 changes: 7 additions & 0 deletions app/Graph/build.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,13 @@

using namespace it_lab_ai;

std::unordered_map<std::string, std::string> model_paths = {
{"alexnet_mnist", MODEL_PATH_H5},
{"googlenet", MODEL_PATH_GOOGLENET_ONNX},
{"resnet", MODEL_PATH_RESNET_ONNX},
{"densenet", MODEL_PATH_DENSENET_ONNX},
{"yolo", MODEL_PATH_YOLO11NET_ONNX}};

Graph build_graph_linear(it_lab_ai::Tensor& input, it_lab_ai::Tensor& output,
bool comments) {
if (comments) {
Expand Down
7 changes: 1 addition & 6 deletions app/Graph/build.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -33,12 +33,7 @@
#include "layers/TransposeLayer.hpp"
#include "layers_oneDNN/EWLayer.hpp"

std::unordered_map<std::string, std::string> model_paths = {
{"alexnet_mnist", MODEL_PATH_H5},
{"googlenet", MODEL_PATH_GOOGLENET_ONNX},
{"resnet", MODEL_PATH_RESNET_ONNX},
{"densenet", MODEL_PATH_DENSENET_ONNX},
{"yolo", MODEL_PATH_YOLO11NET_ONNX}};
extern std::unordered_map<std::string, std::string> model_paths;

struct ParseResult {
std::vector<std::shared_ptr<it_lab_ai::Layer>> layers;
Expand Down
1 change: 0 additions & 1 deletion app/Graph/graph_build.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
#include <numeric>
#include <unordered_map>

#include "build.cpp"
#include "build.hpp"

namespace fs = std::filesystem;
Expand Down
Loading