layer_registry.h

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-only
// Copyright (C) 2026 Artificial Intelligence Techniques, S.L. and Roberto López González
 
#pragma once
 
#include <memory>
#include <string>
 
#include "opennn/core/enum_map.h"
 
namespace opennn
{
 
enum class LayerType
{
    Activation,
    Addition,
    Clamping,
    Concatenation,
    Convolutional,
    Dense,
    Detection,
    DetectionV8,
    Embedding,
    Flatten,
    LSTM,
    MultiHeadAttention,
    Normalization3d,
    GroupedQueryAttention,
    NonMaxSuppression,
    Pooling,
    Pooling3d,
    Recurrent,
    Scaling,
    Tokenizer,
    Unscaling,
    Upsampling,
    C2PSA,
    Count
};
 
class Layer;
 
const EnumMap<LayerType>& layer_type_map();
const std::string& layer_type_to_string(LayerType);
LayerType string_to_layer_type(const std::string&);
 
std::unique_ptr<Layer> create_layer(const std::string& name);
 
}