From: Yangqing Jia Date: Wed, 18 Sep 2013 03:33:21 +0000 (-0700) Subject: naming X-Git-Tag: submit/tizen/20180823.020014~1055 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=5b47cef61cffc78ce5f84782d369e6dc26bb44d5;p=platform%2Fupstream%2Fcaffeonacl.git naming --- diff --git a/src/caffeine/layers/inner_product_layer.cu b/src/caffeine/layers/inner_product_layer.cu index baeca0e..4cae30c 100644 --- a/src/caffeine/layers/inner_product_layer.cu +++ b/src/caffeine/layers/inner_product_layer.cu @@ -101,7 +101,7 @@ Dtype InnerProductLayer::Backward_cpu(const vector*>& top, } template -__global__ void BroadcastCopy(const int total, const int vec_len, +__global__ void BroadcastRow(const int total, const int vec_len, const Dtype* in_vec, Dtype* out_matrix) { int index = threadIdx.x + blockIdx.x * blockDim.x; if (index < total) { @@ -110,6 +110,8 @@ __global__ void BroadcastCopy(const int total, const int vec_len, } } + + template void InnerProductLayer::Forward_gpu(const vector*>& bottom, vector*>* top) { @@ -123,7 +125,7 @@ void InnerProductLayer::Forward_gpu(const vector*>& bottom, beta = 1.; const int count = (*top)[0]->count(); // we pre-copy the bias to the results, and then call gemm. - BroadcastCopy<<>>( + BroadcastRow<<>>( count, N_, bias, top_data); } switch(sizeof(Dtype)) {