diff options
Diffstat (limited to 'src/rnn/cells.h')
-rw-r--r-- | src/rnn/cells.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/rnn/cells.h b/src/rnn/cells.h index cddfd26e..18ac4d1d 100644 --- a/src/rnn/cells.h +++ b/src/rnn/cells.h @@ -197,6 +197,7 @@ public: Expr gruOps(const std::vector<Expr>& nodes, bool final = false); +/** Base class for a gated recurrent unit (GRU) cell. */ class GRU : public Cell { protected: std::string prefix_; |