Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / cl_kernels / convolution_grad_weights_ref.cl

diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/convolution_grad_weights_ref.cl b/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/convolution_grad_weights_ref.cl
index cba96cb..95641e2 100644 (file)
--- a/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/convolution_grad_weights_ref.cl
+++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/convolution_grad_weights_ref.cl
@@ -93,13 +93,16 @@ KERNEL(convolution_grad_weights_gpu_ref)(
 #endif
     }
 
-#if MOMENTUM
-    float update_gradient_w = lr * (grad_w + DECAY_RATE * filter[weights_idx]) + prev_grad_w[weights_idx] * MOMENTUM_FACTOR;
-    filter[weights_idx] -= update_gradient_w;
-    prev_grad_w[weights_idx] = update_gradient_w;
+#if OUTPUT_GRAD_W
+    output[weights_idx] = grad_w;
 #else
-    filter[weights_idx] -= lr * grad_w + DECAY_RATE * lr * filter[weights_idx];
-#endif
+    #if MOMENTUM
+        float update_gradient_w = lr * (grad_w + DECAY_RATE * filter[weights_idx]) + prev_grad_w[weights_idx] * MOMENTUM_FACTOR;
+        filter[weights_idx] -= update_gradient_w;
+        prev_grad_w[weights_idx] = update_gradient_w;
+    #else
+        filter[weights_idx] -= lr * grad_w + DECAY_RATE * lr * filter[weights_idx];
+    #endif
 
 #if BIAS_TERM
         if(ifm == 0 && id_x == 0 && id_y == 0)
@@ -114,4 +117,6 @@ KERNEL(convolution_grad_weights_gpu_ref)(
         }
 #endif
 
+#endif
+
 }