Searched refs:DROPOUT_GEN_MASK (Results 1 – 2 of 2) sorted by relevance
175 if (prim->name() != DROPOUT_GEN_MASK) { in GetDropoutGenMaskPrim()264 Operator replace_op = {std::make_pair(DROPOUT_GEN_MASK, args)}; in GetDropoutGenMaskReplaceOp()
290 constexpr char DROPOUT_GEN_MASK[] = "DropoutGenMask"; variable