EventLayer.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using MyCaffe.common;
using MyCaffe.param;
 
namespace MyCaffe.layers.nt
{
    public class EventLayer<T> : Layer<T>
    {
        public event EventHandler<ForwardArgs<T>> OnLayerSetup;
        public event EventHandler<ForwardArgs<T>> OnReshape;
        public event EventHandler<ForwardArgs<T>> OnForward;
        public event EventHandler<BackwardArgs<T>> OnBackward;
 
        public EventLayer(CudaDnn<T> cuda, Log log, LayerParameter p)
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.EVENT;
        }
 
        public override void LayerSetUp(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            m_bUseHalfSize = m_param.use_halfsize;
 
            if (OnLayerSetup != null)
                OnLayerSetup(this, new ForwardArgs<T>(colBottom, colTop));
        }
 
        public override void Reshape(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            if (OnReshape != null)
            {
                OnReshape(this, new ForwardArgs<T>(colBottom, colTop));
                return;
            }
 
            colTop[0].ReshapeLike(colBottom[0], m_bUseHalfSize);
        }
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            if (OnForward != null)
            {
                OnForward(this, new ForwardArgs<T>(colBottom, colTop));
                return;
            }
 
            for (int i = 0; i < colBottom.Count && i < colTop.Count; i++)
            {
                int nCount = colTop[i].count();
                int nCountB = colBottom[i].count();
 
                m_log.CHECK_EQ(nCount, nCountB, "The top and bottom at " + i.ToString() + " must have the same number of items.");
 
                long hBottomData = colBottom[i].gpu_data;
                long hTopData = colTop[i].mutable_gpu_data;
 
                m_cuda.copy(nCount, hBottomData, hTopData);
            }
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
            if (OnBackward != null)
            {
                OnBackward(this, new BackwardArgs<T>(colTop, rgbPropagateDown, colBottom));
                return;
            }
 
            for (int i = 0; i < colBottom.Count && i < colTop.Count; i++)
            {
                if (rgbPropagateDown[i])
                {
                    int nCount = colTop[i].count();
                    int nCountB = colBottom[i].count();
 
                    m_log.CHECK_EQ(nCount, nCountB, "The top and bottom at " + i.ToString() + " must have the same number of items.");
 
                    long hBottomDiff = colBottom[i].mutable_gpu_diff;
                    long hTopDiff = colTop[i].gpu_diff;
 
                    m_cuda.copy(nCount, hTopDiff, hBottomDiff);
                }
            }
        }
    }
}