Home
last modified time | relevance | path

Searched refs:seeds_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drandom_dataset_op.cc41 : DatasetBase(DatasetContext(ctx)), seeds_(seed, seed2) {} in Dataset()
61 return strings::StrCat("RandomDatasetOp(", seeds_.first, ", ", in DebugString()
62 seeds_.second, ")::Dataset"); in DebugString()
79 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.first, &seed)); in AsGraphDefInternal()
80 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.second, &seed2)); in AsGraphDefInternal()
90 seeds_(MaybeOverrideSeeds(dataset()->seeds_)), in Iterator()
91 parent_generator_(seeds_.first, seeds_.second), in Iterator()
124 parent_generator_ = random::PhiloxRandom(seeds_.first, seeds_.second); in RestoreInternal()
138 const std::pair<int64, int64> seeds_; member in tensorflow::data::experimental::RandomDatasetOp::Dataset::Iterator
146 const std::pair<int64, int64> seeds_; member in tensorflow::data::experimental::RandomDatasetOp::Dataset
Dsampling_dataset_op.cc47 seeds_(seed, seed2), in Dataset()
58 seeds_.first, seeds_.second)); in MakeIteratorInternal()
92 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.first, &seed)); in AsGraphDefInternal()
93 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.second, &seed2)); in AsGraphDefInternal()
104 seeds_(MaybeOverrideSeeds({seed, seed2})), in Iterator()
105 parent_generator_(seeds_.first, seeds_.second),
147 parent_generator_ = random::PhiloxRandom(seeds_.first, seeds_.second); in ResetRngs()
160 writer->WriteScalar(this->full_name("seed"), seeds_.first)); in SaveInternal()
162 writer->WriteScalar(this->full_name("seed2"), seeds_.second)); in SaveInternal()
183 seeds_ = {seed, seed2}; in RestoreInternal()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Drandom_seed_ops.h94 explicit FixedSeedGenerator(RandomSeeds seeds) : seeds_(std::move(seeds)) {} in FixedSeedGenerator()
96 int64 seed() const override { return seeds_.seed(); } in seed()
97 int64 seed2() const override { return seeds_.seed(); } in seed2()
104 const RandomSeeds seeds_;
111 : seeds_(std::move(seeds)), in RandomSeedGenerator()
112 parent_generator_(seeds_.seed(), seeds_.seed2()), in RandomSeedGenerator()
115 int64 seed() const override { return seeds_.seed(); } in seed()
116 int64 seed2() const override { return seeds_.seed2(); } in seed2()
123 const RandomSeeds seeds_;
146 std::unique_ptr<RandomSeeds> seeds_ TF_GUARDED_BY(mu_);
Drandom_seed_ops.cc44 *seed1 = seeds_.seed(); in GenerateSeeds()
45 *seed2 = seeds_.seed2(); in GenerateSeeds()
59 parent_generator_ = random::PhiloxRandom(seeds_.seed(), seeds_.seed2()); in Reset()
76 seeds_ = absl::make_unique<RandomSeeds>(seed, seed2); in Compute()
89 *manager = new SeedGeneratorManager(new RandomSeedGenerator(*seeds_)); in CreateResource()
91 *manager = new SeedGeneratorManager(new FixedSeedGenerator(*seeds_)); in CreateResource()
93 seeds_ = nullptr; in CreateResource()
Dshuffle_dataset_op.cc450 seeds_(std::move(seeds)) {} in Dataset()
475 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
476 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed2(), &seed2_node)); in AsGraphDefInternal()
492 const RandomSeeds seeds_; member in tensorflow::data::ShuffleDatasetOp::Dataset
563 seeds_(std::move(seeds)) {} in DatasetV3()
588 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
589 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed2(), &seed2_node)); in AsGraphDefInternal()
612 const RandomSeeds seeds_; member in tensorflow::data::ShuffleDatasetOp::DatasetV3
756 seeds_(std::move(seeds)) {} in Dataset()
781 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed)); in AsGraphDefInternal()
[all …]