Searched defs:CellParams (Results 1 – 2 of 2) sorted by relevance
175 struct CellParams : public CellParamsBase { struct176 CellParams( in CellParams() argument184 const Tensor& w_ih;185 const Tensor& w_hh;186 const Tensor& b_ih_; /* optional */187 const Tensor& b_hh_; /* optional */188 const Tensor& w_hr; /* only defined for LSTMs with projections */190 Tensor matmul_ih(const Tensor& input) const override { in matmul_ih()193 Tensor matmul_hh(const Tensor& h) const override { in matmul_hh()196 Tensor matmul_hr(const Tensor& h) const override { in matmul_hr()[all …]
2458 "CellParams":{ object7839 "CellParams":{ object7863 "CellParams":{ object