From 08d0e07ed64139d2552d7801b755af7f34b4a75e Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Sun, 19 Aug 2018 11:08:02 +0000
Subject: [PATCH] Update Readme.md
---
src/blas.c | 32 ++++++++++++++++----------------
1 files changed, 16 insertions(+), 16 deletions(-)
diff --git a/src/blas.c b/src/blas.c
index 4ff0b83..ccf0522 100644
--- a/src/blas.c
+++ b/src/blas.c
@@ -11,8 +11,8 @@
int b,i,j,k;
int in_c = out_c/(stride*stride);
- //printf("\n out_c = %d, out_w = %d, out_h = %d, stride = %d, forward = %d \n", out_c, out_w, out_h, stride, forward);
- //printf(" in_c = %d, in_w = %d, in_h = %d \n", in_c, out_w*stride, out_h*stride);
+ //printf("\n out_c = %d, out_w = %d, out_h = %d, stride = %d, forward = %d \n", out_c, out_w, out_h, stride, forward);
+ //printf(" in_c = %d, in_w = %d, in_h = %d \n", in_c, out_w*stride, out_h*stride);
for(b = 0; b < batch; ++b){
for(k = 0; k < out_c; ++k){
@@ -24,7 +24,7 @@
int w2 = i*stride + offset % stride;
int h2 = j*stride + offset / stride;
int out_index = w2 + out_w*stride*(h2 + out_h*stride*(c2 + in_c*b));
- if(forward) out[out_index] = x[in_index]; // used by default for forward (i.e. forward = 0)
+ if(forward) out[out_index] = x[in_index]; // used by default for forward (i.e. forward = 0)
else out[in_index] = x[out_index];
}
}
@@ -293,17 +293,17 @@
void upsample_cpu(float *in, int w, int h, int c, int batch, int stride, int forward, float scale, float *out)
{
- int i, j, k, b;
- for (b = 0; b < batch; ++b) {
- for (k = 0; k < c; ++k) {
- for (j = 0; j < h*stride; ++j) {
- for (i = 0; i < w*stride; ++i) {
- int in_index = b*w*h*c + k*w*h + (j / stride)*w + i / stride;
- int out_index = b*w*h*c*stride*stride + k*w*h*stride*stride + j*w*stride + i;
- if (forward) out[out_index] = scale*in[in_index];
- else in[in_index] += scale*out[out_index];
- }
- }
- }
- }
+ int i, j, k, b;
+ for (b = 0; b < batch; ++b) {
+ for (k = 0; k < c; ++k) {
+ for (j = 0; j < h*stride; ++j) {
+ for (i = 0; i < w*stride; ++i) {
+ int in_index = b*w*h*c + k*w*h + (j / stride)*w + i / stride;
+ int out_index = b*w*h*c*stride*stride + k*w*h*stride*stride + j*w*stride + i;
+ if (forward) out[out_index] = scale*in[in_index];
+ else in[in_index] += scale*out[out_index];
+ }
+ }
+ }
+ }
}
\ No newline at end of file
--
Gitblit v1.10.0