Searched refs:kPrimDropoutGenMask (Results 1 – 8 of 8) sorted by relevance
33 DropoutGenMask() : PrimitiveC(prim::kPrimDropoutGenMask->name()) {} in DropoutGenMask()
158 REGISTER_PRIMITIVE_EVAL_IMPL(DropoutGenMask, prim::kPrimDropoutGenMask, DropoutGenMaskInfer, nullpt…
25 REG_ADPT_DESC(DropOutGenMask, prim::kPrimDropoutGenMask->name(), ADPT_DESC(DropOutGenMask))
101 EXPECT_EQ(AnfAlgo::GetCNodeName(cnode), prim::kPrimDropoutGenMask->name()); in TEST_F()
44 if (AnfAlgo::GetCNodeName(cnode) != prim::kPrimDropoutGenMask->name()) { in Process()
44 const auto kDropoutGenMask = prim::kPrimDropoutGenMask->name(); in GetDependsFormMap()
36 …static std::unordered_set<PrimitivePtr> not_recomputed_op_list{prim::kPrimDropoutGenMask, prim::kP… in CanNotRecomputed()
356 inline const PrimitivePtr kPrimDropoutGenMask = std::make_shared<Primitive>(kDropoutGenMask); variable