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.nt
{
    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.EVENT:
                    return new EventLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.GRAM:
                    return new GramLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.ONEHOT:
                    return new OneHotLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.SCALAR:
                    return new ScalarLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.TV_LOSS:
                    return new TVLossLayer<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.EVENT:
                    return new EventLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.GRAM:
                    return new GramLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.ONEHOT:
                    return new OneHotLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.SCALAR:
                    return new ScalarLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.TV_LOSS:
                    return new TVLossLayer<float>(cuda, log, p);
 
                default:
                    return null;
            }
        }
    }
}