Pooling.h

// Copyright (C) Entropy Software LLC - All Rights Reserved
 
#pragma once
 
#include "GraphNode.h"
 
namespace SmartEngine
{
 
class IPoolingLayer;
 
#pragma pack(push, 4)
enum class PoolType : byte
{
    Max,
 
    Average,
 
    Count,
};
SMARTENGINE_ENUM_STRING(PoolType, "Max", "Average");
 
struct PoolingLayerCInfo : public GraphNodeCInfo
{
    IGraphNode* input = nullptr;
 
    PoolType type = PoolType::Max;
 
    int inputWidth = -1;
 
    int inputHeight = -1;
};
 
struct UnpoolingLayerCInfo : public GraphNodeCInfo
{
    IGraphNode* input = nullptr;
 
    IPoolingLayer* poolingLayer = nullptr;
};
#pragma pack(pop)
 
class SMARTENGINE_EXPORT IPoolingLayer : public IGraphNode
{
public:
    SMARTENGINE_DECLARE_CLASS(IPoolingLayer)
};
 
class SMARTENGINE_EXPORT IUnpoolingLayer : public IGraphNode
{
public:
    SMARTENGINE_DECLARE_CLASS(IUnpoolingLayer)
};
 
SMARTENGINE_EXPORT ObjectPtr<IPoolingLayer> CreatePoolingLayer(const PoolingLayerCInfo& cinfo);
 
SMARTENGINE_EXPORT ObjectPtr<IUnpoolingLayer> CreateUnpoolingLayer(const UnpoolingLayerCInfo& cinfo);
 
extern "C"
{
    SMARTENGINE_EXPORT ObjPtr PoolingLayer_CreateInstance(const PoolingLayerCInfo& cinfo);
    SMARTENGINE_EXPORT ObjPtr UnpoolingLayer_CreateInstance(const UnpoolingLayerCInfo& cinfo);
}
 
} // namespace SmartEngine