Searched defs:GRUCell (Results 1 – 3 of 3) sorted by relevance
30 struct GRUCell { struct31 GRUCell(const int batch_size, const int input_size, const int cell_size) in GRUCell() function36 inline Eigen::array<Eigen::DenseIndex, 2> x_offsets() const { return {0, 0}; } in x_offsets()38 inline Eigen::array<Eigen::DenseIndex, 2> x_extends() const { in x_extends()42 inline Eigen::array<Eigen::DenseIndex, 2> h_offsets() const { in h_offsets()69 struct GRUBlockCellFprop : public GRUCell { argument
481 class GRUCell(LayerRNNCell): class
1515 class GRUCell(DropoutRNNCellMixin, Layer): class