perseusdg
41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
#include <iostream>
|
|
|
|
#include "Layer.h"
|
|
#include "kernels.h"
|
|
|
|
namespace tk { namespace dnn {
|
|
|
|
Shortcut::Shortcut(Network *net, Layer *backLayer, bool mul) : Layer(net) {
|
|
|
|
this->backLayer = backLayer;
|
|
this->mul = mul;
|
|
this->c = input_dim.c;
|
|
this->h = input_dim.h;
|
|
this->w = input_dim.w;
|
|
checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
|
|
|
|
if( ( backLayer->output_dim.c != input_dim.c && mul ) ||
|
|
(( backLayer->output_dim.w != input_dim.w || backLayer->output_dim.h != input_dim.h ) && !mul ) )
|
|
FatalError("Shortcut dim missmatch");
|
|
|
|
}
|
|
|
|
Shortcut::~Shortcut() {
|
|
|
|
checkCuda( cudaFree(dstData) );
|
|
}
|
|
|
|
dnnType* Shortcut::infer(dataDim_t &dim, dnnType* srcData) {
|
|
|
|
dataDim_t bdim = this->backLayer->output_dim;
|
|
|
|
checkCuda(cudaMemcpy(dstData, srcData, dim.tot()*sizeof(dnnType), cudaMemcpyDeviceToDevice));
|
|
shortcutForward(this->backLayer->dstData, dstData, dim.n, dim.c, dim.h, dim.w, 1, bdim.n, bdim.c, bdim.h, bdim.w, 1, mul);
|
|
|
|
//update data dimensions
|
|
dim = output_dim;
|
|
|
|
return dstData;
|
|
}
|
|
|
|
}} |