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.cc46 : DatasetBase(DatasetContext(ctx)), seeds_(seed, seed2) {} in Dataset()
76 return strings::StrCat("RandomDatasetOp(", seeds_.first, ", ", in DebugString()
77 seeds_.second, ")::Dataset"); in DebugString()
94 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.first, &seed)); in AsGraphDefInternal()
95 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.second, &seed2)); in AsGraphDefInternal()
105 seeds_(MaybeOverrideSeeds(dataset()->seeds_)), in Iterator()
106 parent_generator_(seeds_.first, seeds_.second), in Iterator()
139 parent_generator_ = random::PhiloxRandom(seeds_.first, seeds_.second); in RestoreInternal()
153 const std::pair<int64_t, int64_t> seeds_; member in tensorflow::data::experimental::RandomDatasetOp::Dataset::Iterator
161 const std::pair<int64_t, int64_t> 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_t seed() const override { return seeds_.seed(); } in seed()
97 int64_t 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_t seed() const override { return seeds_.seed(); } in seed()
116 int64_t 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.cc42 *seed1 = seeds_.seed(); in GenerateSeeds()
43 *seed2 = seeds_.seed2(); in GenerateSeeds()
57 parent_generator_ = random::PhiloxRandom(seeds_.seed(), seeds_.seed2()); in Reset()
76 seeds_ = std::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.cc528 seeds_(std::move(seeds)) {} in Dataset()
553 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
554 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed2(), &seed2_node)); in AsGraphDefInternal()
570 const RandomSeeds seeds_; member in tensorflow::data::ShuffleDatasetOp::Dataset
641 seeds_(std::move(seeds)) {} in DatasetV3()
666 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
667 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed2(), &seed2_node)); in AsGraphDefInternal()
690 const RandomSeeds seeds_; member in tensorflow::data::ShuffleDatasetOp::DatasetV3
834 seeds_(std::move(seeds)) {} in Dataset()
859 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed)); in AsGraphDefInternal()
[all …]