RNN cell base class.

src/ngraph/op/fused/rnn_cell_base.cpp

+//*****************************************************************************
+// Copyright 2017-2019 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//*****************************************************************************
+
+#include <algorithm>
+#include <iterator>
+
+#include "ngraph/op/fused/rnn_cell_base.hpp"
+#include "ngraph/util.hpp"
+
+using namespace std;
+using namespace ngraph;
+
+// Modify input vector in-place and return reference to modified vector.
+static vector<string> to_lower_case(const vector<string>& vs)
+{
+    vector<string> res(vs);
+    transform(begin(res), end(res), begin(res), [](string& s) { return to_lower(s); });
+    return res;
+}
+
+op::RNNCellBase::RNNCellBase(size_t hidden_size,
+                             float clip,
+                             const vector<string>& activations,
+                             const vector<float>& activation_alpha,
+                             const vector<float>& activation_beta)
+    : m_hidden_size(hidden_size)
+    , m_clip(clip)
+    , m_activations(to_lower_case(activations))
+    , m_activation_alpha(activation_alpha)
+    , m_activation_beta(activation_beta)
+{
+}
+
+op::ActivationFunction op::RNNCellBase::get_activation_function(size_t idx) const
+{
+    op::ActivationFunction afunc = get_activation_func_by_name(m_activations.at(idx));
+
+    // Set activation functions parameters (if any)
+    if (m_activation_alpha.size() > idx)
+    {
+        afunc.set_alpha(m_activation_alpha.at(idx));
+    }
+    if (m_activation_beta.size() > idx)
+    {
+        afunc.set_beta(m_activation_beta.at(idx));
+    }
+
+    return afunc;
+}

src/ngraph/op/fused/rnn_cell_base.hpp

+//*****************************************************************************
+// Copyright 2017-2019 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//*****************************************************************************
+
+#pragma once
+
+#include <cstddef>
+#include <string>
+#include <vector>
+
+#include "ngraph/op/util/activation_functions.hpp"
+
+namespace ngraph
+{
+    namespace op
+    {
+        /// \brief      Base class for all recurrent network cells.
+        ///
+        /// \note       It holds all common attributes.
+        ///
+        class RNNCellBase
+        {
+        public:
+            ///
+            /// \brief      Constructs a RNNCellBase class.
+            ///
+            /// \param[in]  hidden_size       The number of hidden units for recurrent cell.
+            /// \param[in]  clip              The value defining clipping range [-clip, clip] on
+            ///                               input of activation functions.
+            /// \param[in]  activations       The vector of activation functions used inside
+            ///                               recurrent cell.
+            /// \param[in]  activation_alpha  The vector of alpha parameters for activation
+            ///                               functions in order respective to activation list.
+            /// \param[in]  activation_beta   The vector of beta parameters for activation
+            ///                               functions in order respective to activation list.
+            ///
+            RNNCellBase(std::size_t hidden_size,
+                        float clip,
+                        const std::vector<std::string>& activations,
+                        const std::vector<float>& activation_alpha,
+                        const std::vector<float>& activation_beta);
+
+            std::size_t get_hidden_size() const { return m_hidden_size; };
+            float get_clip() const { return m_clip; };
+            const std::vector<std::string>& get_activations() const { return m_activations; };
+            const std::vector<float>& get_activation_alpha() const { return m_activation_alpha; };
+            const std::vector<float>& get_activation_beta() const { return m_activation_beta; };
+        protected:
+            ///
+            /// \brief      Constructs activation function object.
+            ///
+            /// \param[in]  idx   The index of the activation function name.
+            ///
+            /// \return     The object representing activation function.
+            ///
+            ActivationFunction get_activation_function(std::size_t idx) const;
+
+        private:
+            std::size_t m_hidden_size = 0.f;
+            float m_clip = 0.f;
+            std::vector<std::string> m_activations;
+            std::vector<float> m_activation_alpha;
+            std::vector<float> m_activation_beta;
+        };
+    }
+}