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.tft
{
    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.DATA_TEMPORAL:
                    return new DataTemporalLayer<double>(cuda, log, p, evtCancel, db);
 
                case LayerParameter.LayerType.CATEGORICAL_TRANS:
                    return new CategoricalTransformationLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.CHANNEL_EMBEDDING:
                    return new ChannelEmbeddingLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.GATEADDNORM:
                    return new GateAddNormLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.GLU:
                    return new GluLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.GRN:
                    return new GrnLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.NUMERIC_TRANS:
                    return new NumericTransformationLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.MULTIHEAD_ATTENTION_INTERP:
                    return new MultiHeadAttentionInterpLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.RESHAPE_TEMPORAL:
                    return new ReshapeTemporalLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.VARSELNET:
                    return new VarSetNetLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.QUANTILE_ACCURACY:
                    return new QuantileAccuracyLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.QUANTILE_LOSS:
                    return new QuantileLossLayer<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.DATA_TEMPORAL:
                    return new DataTemporalLayer<float>(cuda, log, p, evtCancel, db);
 
                case LayerParameter.LayerType.CATEGORICAL_TRANS:
                    return new CategoricalTransformationLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.CHANNEL_EMBEDDING:
                    return new ChannelEmbeddingLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.GATEADDNORM:
                    return new GateAddNormLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.GLU:
                    return new GluLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.GRN:
                    return new GrnLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.NUMERIC_TRANS:
                    return new NumericTransformationLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.MULTIHEAD_ATTENTION_INTERP:
                    return new MultiHeadAttentionInterpLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.RESHAPE_TEMPORAL:
                    return new ReshapeTemporalLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.VARSELNET:
                    return new VarSetNetLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.QUANTILE_ACCURACY:
                    return new QuantileAccuracyLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.QUANTILE_LOSS:
                    return new QuantileLossLayer<float>(cuda, log, p);
 
                default:
                    return null;
            }
        }
    }
}