Searched refs:kPrimLstm (Results 1 – 3 of 3) sorted by relevance
43 if (!opt::CheckPrimitiveType(cnode, prim::kPrimLstm)) { in GetRealLstmWeightNode()254 if (CheckPrimitiveType(node, prim::kPrimLstm)) { in Run()268 if (!CheckPrimitiveType(cnode->input(1)->cast<CNodePtr>(), prim::kPrimLstm)) { in Run()
434 } else if (opt::CheckPrimitiveType(cnode, prim::kPrimLstm)) { in DoQuantize()
269 inline const PrimitivePtr kPrimLstm = std::make_shared<Primitive>("LSTM"); variable