Fix retrieving bias input.

src/ngraph/op/fused/gru_cell.cpp

@@ -191,7 +191,7 @@ NodeVector op::GRUCell::decompose_op() const
     std::shared_ptr<Node> W = get_argument(1);
     std::shared_ptr<Node> R = get_argument(2);
     std::shared_ptr<Node> H_t = get_argument(3);
-    std::shared_ptr<Node> B = get_bias();
+    std::shared_ptr<Node> B = get_argument(4);
 
     // Get W and R biases separately.
     NodeVector b_W_R = builder::split(B, 2);
@@ -270,15 +270,6 @@ NodeVector op::GRUCell::decompose_op() const
     return {H_t};
 }
 
-shared_ptr<Node> op::GRUCell::get_bias() const
-{
-    shared_ptr<Node> bias;
-    // Split B onto Wb an Rb and add them.
-    NodeVector b_W_R = builder::split(get_argument(4), 2);
-    bias = b_W_R.at(0) + b_W_R.at(1);
-    return bias;
-}
-
 void op::GRUCell::add_default_bias_input()
 {
     shared_ptr<Node> B =

src/ngraph/op/fused/gru_cell.hpp

@@ -132,8 +132,6 @@ namespace ngraph
 
             bool get_linear_before_reset() const { return m_linear_before_reset; }
         private:
-            std::shared_ptr<Node> get_bias() const;
-
             /// brief Add and initialize bias input to all zeros.
             void add_default_bias_input();
 

src/ngraph/op/fused/rnn_cell.cpp

@@ -171,10 +171,7 @@ NodeVector op::RNNCell::decompose_op() const
     std::shared_ptr<Node> W = get_argument(1);
     std::shared_ptr<Node> R = get_argument(2);
     std::shared_ptr<Node> H_t = get_argument(3);
-    std::shared_ptr<Node> B = get_bias();
-
-    NodeVector b_W_R = builder::split(B, 2);
-    auto bias = b_W_R.at(0) + b_W_R.at(1);
+    std::shared_ptr<Node> bias = get_bias();
 
     // Xt*(W^T)
     auto Xt_W = std::make_shared<op::Dot>(X, builder::transpose(W));