LayerFactory.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using MyCaffe.basecode;
using MyCaffe.common;
using MyCaffe.db.image;
using MyCaffe.param;
 
namespace MyCaffe.layers.lnn
{
    public class LayerFactory : ILayerCreator
    {
        public Layer<double> CreateDouble(CudaDnn<double> cuda, Log log, LayerParameter p, CancelEvent evtCancel, IXDatabaseBase db)
        {
            switch (p.type)
            {
                case LayerParameter.LayerType.CFC:
                    return new CfcLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.CFC_UNIT:
                    return new CfcUnitLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.LTC_UNIT:
                    return new LtcUnitLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.LECUN:
                    return new LeCunLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.SILU:
                    return new SiLULayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.SOFTPLUS:
                    return new SoftPlusLayer<double>(cuda, log, p);
 
                default:
                    return null;
            }
        }
 
        public Layer<float> CreateSingle(CudaDnn<float> cuda, Log log, LayerParameter p, CancelEvent evtCancel, IXDatabaseBase db)
        {
            switch (p.type)
            {
                case LayerParameter.LayerType.CFC:
                    return new CfcLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.CFC_UNIT:
                    return new CfcUnitLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.LTC_UNIT:
                    return new LtcUnitLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.LECUN:
                    return new LeCunLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.SILU:
                    return new SiLULayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.SOFTPLUS:
                    return new SoftPlusLayer<float>(cuda, log, p);
 
                default:
                    return null;
            }
        }
    }
}