Publishing 2019 R1 content

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

diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_elt_kernel_base.cpp b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_elt_kernel_base.cpp
index 6170abd..26fdb93 100644 (file)
--- a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_elt_kernel_base.cpp
+++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/lstm/lstm_elt_kernel_base.cpp
@@ -26,7 +26,11 @@ namespace kernel_selector
 
         if (params.has_cell) {
             const auto& cell = params.cell;
-            jit.AddConstants({ MakeJitConstant("CELL_TERM", true), MakeJitConstant("CELL", cell) });
+            jit.AddConstants({
+                MakeJitConstant("CELL_TERM", true),
+                MakeJitConstant("CELL", cell),
+                MakeJitConstant("CELL_DIRECTION", params.cell_direction)
+            });
         }
         if (params.clip > 0) {
             std::string psclip = toCodeString(params.clip);
@@ -40,6 +44,7 @@ namespace kernel_selector
         if (params.input_forget) {
             jit.AddConstants({ MakeJitConstant("INPUT_FORGET", true) });
         }
+        jit.AddConstants({ MakeJitConstant("DIRECTION", params.direction) });
 
         const auto& GEMMInput = params.inputs[0];
         size_t size = GEMMInput.X().v / 4;