| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 76 seed2_(seed2), in Iterator()
190 parent_generator_ = random::PhiloxRandom(seed_, seed2_); in ResetRngs()
203 writer->WriteScalar(this->full_name("seed2"), seed2_)); in SaveInternal()
252 reader->ReadScalar(this->full_name("seed2"), &seed2_)); in RestoreInternal()
306 int64 seed2_ GUARDED_BY(mu_);
394 seed2_(seed2) {} in ReshufflingDataset()
398 ", ", seed2_, ")::ReshufflingDataset"); in DebugString()
405 seed2_); in MakeIteratorInternal()
413 seed2_(seed2), in RandomSeedGenerator()
442 parent_generator_ = random::PhiloxRandom(seed_, seed2_); in Reset()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | random_dataset_op.cc | 55 : DatasetBase(DatasetContext(ctx)), seed_(seed), seed2_(seed2) {} in Dataset()
75 return strings::StrCat("RandomDatasetOp(", seed_, ", ", seed2_, in DebugString()
88 TF_RETURN_IF_ERROR(b->AddScalar(seed2_, &seed2)); in AsGraphDefInternal()
98 parent_generator_(dataset()->seed_, dataset()->seed2_), in Iterator()
131 random::PhiloxRandom(dataset()->seed_, dataset()->seed2_); in RestoreInternal()
153 const int64 seed2_; member in tensorflow::data::__anonc28f42b70111::RandomDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fractional_avg_pool_op.cc | 52 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in FractionalAvgPoolOp()
55 if ((seed_ == 0) && (seed2_ == 0)) { in FractionalAvgPoolOp()
57 seed2_ = random::New64(); in FractionalAvgPoolOp()
61 context, (seed_ == 0) && (seed2_ == 0), in FractionalAvgPoolOp()
95 generator.Init(seed_, seed2_); in Compute()
186 int64 seed2_; member in tensorflow::FractionalAvgPoolOp
|
| D | fractional_max_pool_op.cc | 55 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in FractionalMaxPoolOp()
58 if ((seed_ == 0) && (seed2_ == 0)) { in FractionalMaxPoolOp()
60 seed2_ = random::New64(); in FractionalMaxPoolOp()
64 context, (seed_ == 0) && (seed2_ == 0), in FractionalMaxPoolOp()
100 generator.Init(seed_, seed2_); in Compute()
187 int64 seed2_; member in tensorflow::FractionalMaxPoolOp
|
| D | random_shuffle_queue_op.cc | 489 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in RandomShuffleQueueOp()
498 capacity_, min_after_dequeue_, seed_, seed2_, component_types_, in CreateResource()
505 int64 seed2_; member in tensorflow::RandomShuffleQueueOp
|
| D | cudnn_rnn_ops.cc | 907 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in CudnnRNNKernelCommon()
931 uint64 seed() { return (static_cast<uint64>(seed_) << 32) | seed2_; } in seed()
1024 int seed2_; member in tensorflow::CudnnRNNKernelCommon
|