Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / actual_kernels / lstm / lstm_gemm_kernel_base.cpp

diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_gemm_kernel_base.cpp b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_gemm_kernel_base.cpp
index 7030085..a684643 100644 (file)
--- a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_gemm_kernel_base.cpp
+++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_gemm_kernel_base.cpp
@@ -31,11 +31,15 @@ namespace kernel_selector
             jit.AddConstants({ MakeJitConstant("BIAS", bias), MakeJitConstant("BIAS_TERM", true) });
         }
         if (params.hasHidden) {
-            jit.AddConstants({ MakeJitConstant("HIDDEN", hidden), MakeJitConstant("HIDDEN_TERM", true) , MakeJitConstant("RECURRENT", recurrent) });
+            jit.AddConstants({ MakeJitConstant("HIDDEN", hidden),
+                MakeJitConstant("HIDDEN_TERM", true),
+                MakeJitConstant("RECURRENT", recurrent),
+                MakeJitConstant("HIDDEN_DIRECTION", params.hidden_direction)
+            });
         }
-
         jit.AddConstants({ MakeJitConstant("WEIGHTS", weights)});
         jit.AddConstants({ MakeJitConstant("DIRECTION", params.direction)});
+        jit.AddConstants({ MakeJitConstant("INPUT_DIRECTION", params.input_direction)});
 
         return jit;
     }
@@ -51,7 +55,7 @@ namespace kernel_selector
 
         KernelData kd = KernelData::Default<lstm_gemm_params>(params, orgParams.inputs.size());
 
-        float effiency = FORCE_PRIORITY_1;
+        float effiency = FORCE_PRIORITY_9;
         const auto& input = orgParams.inputs[0];
 
         auto newParams = orgParams;