/external/tensorflow/tensorflow/core/kernels/data/ |
D | shuffle_dataset_op.cc | 70 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 …]
|
D | random_dataset_op.cc | 55 : 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/ |
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
|
/external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
D | cudnn_rnn_ops.cc | 491 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
|