Home
last modified time | relevance | path

Searched refs:seed2_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc70 seed2_(seed2), in Iterator()
280 parent_generator_ = random::PhiloxRandom(seed_, seed2_); in ResetRngs()
290 const int64 seed2_ GUARDED_BY(mu_);
355 seed2_(seed2), in ReshufflingDataset()
361 ", ", seed2_, ")::ReshufflingDataset"); in DebugString()
380 const int64 seed2_; member in tensorflow::__anon090660da0111::ShuffleDatasetOp::ReshufflingDataset
395 seed2_(seed) {} in FixedSeedDataset()
399 ", ", seed2_, ")::FixedSeedDataset"); in DebugString()
405 {this, strings::StrCat(prefix, "::Shuffle")}, seed_, seed2_)); in MakeIterator()
420 TF_RETURN_IF_ERROR(b->AddScalar(seed2_, &seed2)); in AsGraphDefInternal()
[all …]
Drandom_dataset_op.cc55 : GraphDatasetBase(ctx), seed_(seed), seed2_(seed2) {} in Dataset()
75 return strings::StrCat("RandomDatasetOp(", seed_, ", ", seed2_, in DebugString()
85 TF_RETURN_IF_ERROR(b->AddScalar(seed2_, &seed2)); in AsGraphDefInternal()
95 parent_generator_(dataset()->seed_, dataset()->seed2_), in Iterator()
123 random::PhiloxRandom(dataset()->seed_, dataset()->seed2_); in RestoreInternal()
145 const int64 seed2_; member in tensorflow::__anon3934b98e0111::RandomDatasetOp::Dataset
/external/tensorflow/tensorflow/core/kernels/
Dfractional_avg_pool_op.cc52 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
Dfractional_max_pool_op.cc55 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
Drandom_shuffle_queue_op.cc489 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
/external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/
Dcudnn_rnn_ops.cc491 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in CudnnRNNKernelCommon()
515 uint64 seed() { return (static_cast<uint64>(seed_) << 32) | seed2_; } in seed()
560 int seed2_; member in tensorflow::CudnnRNNKernelCommon