Searched refs:SEED0 (Results 1 – 3 of 3) sorted by relevance
240 if ((attr.find(SEED0) == attr.end()) || (attr.find(SEED1) == attr.end())) { in GetDropoutGenMaskReplaceOp()245 int64_t seed_0 = GetValue<int64_t>(attr[SEED0]); in GetDropoutGenMaskReplaceOp()257 Attr attr_0 = std::make_pair(SEED0, MakeValue(seed_0)); in GetDropoutGenMaskReplaceOp()
259 auto iter0 = attrs_.find(SEED0); in GetAttrs()315 Attr attr_seed0 = std::make_pair(SEED0, new_seed0); in InferReplaceOps()
180 constexpr char SEED0[] = "Seed0"; variable