#include "GraphBuilderRegistry.h"
-#include "Op/Concatenation.h"
-#include "Op/Convolution.h"
-#include "Op/Eltwise.h"
-#include "Op/Input.h"
-#include "Op/Pooling.h"
-#include "Op/ReLU.h"
-#include "Op/Scale.h"
+#include "Layer/Concatenation.h"
+#include "Layer/Convolution.h"
+#include "Layer/Eltwise.h"
+#include "Layer/Input.h"
+#include "Layer/Pooling.h"
+#include "Layer/ReLU.h"
+#include "Layer/Scale.h"
#include <stdex/Memory.h>
class GraphBuilderRegistry
{
public:
- const GraphBuilder *lookup(std::string op) const
+ const GraphBuilder *lookup(std::string layer) const
{
- if (_builder_map.find(op) == _builder_map.end())
+ if (_builder_map.find(layer) == _builder_map.end())
return nullptr;
- return _builder_map.at(op).get();
+ return _builder_map.at(layer).get();
}
static GraphBuilderRegistry &get()