Home | History | Annotate | Download | only in data

Lines Matching defs:seed2

38     int64 seed2;
39     OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed2", &seed2));
43     if (seed == 0 && seed2 == 0) {
45       seed2 = random::New64();
48     *output = new Dataset(ctx, seed, seed2);
54     Dataset(OpKernelContext* ctx, int64 seed, int64 seed2)
55         : GraphDatasetBase(ctx), seed_(seed), seed2_(seed2) {}
83       Node* seed2 = nullptr;
85       TF_RETURN_IF_ERROR(b->AddScalar(seed2_, &seed2));
86       TF_RETURN_IF_ERROR(b->AddDataset(this, {seed, seed2}, output));