Searched defs:LSTMBlockCell (Results 1 – 2 of 2) sorted by relevance
94 struct LSTMBlockCell { struct95 LSTMBlockCell(const int batch_size, const int input_size, const int cell_size) in LSTMBlockCell() function100 int batch_size() const { return batch_size_; } in batch_size()102 int input_size() const { return input_size_; } in input_size()104 int cell_size() const { return cell_size_; } in cell_size()106 inline Eigen::array<Eigen::DenseIndex, 2> icfo_i_offsets() const { in icfo_i_offsets()110 inline Eigen::array<Eigen::DenseIndex, 2> icfo_c_offsets() const { in icfo_c_offsets()114 inline Eigen::array<Eigen::DenseIndex, 2> icfo_f_offsets() const { in icfo_f_offsets()118 inline Eigen::array<Eigen::DenseIndex, 2> icfo_o_offsets() const { in icfo_o_offsets()122 inline Eigen::array<Eigen::DenseIndex, 2> cell_extents() const { in cell_extents()[all …]
335 class LSTMBlockCell(LayerRNNCell): class