| /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 | 490 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in RandomShuffleQueueOp()
499 capacity_, min_after_dequeue_, seed_, seed2_, component_types_, in CreateResource()
506 int64 seed2_; member in tensorflow::RandomShuffleQueueOp
|
| D | cudnn_rnn_ops.cc | 981 OP_REQUIRES_OK(context, context->GetAttr("seed2", &seed2_)); in CudnnRNNKernelCommon()
1005 uint64 seed() { return (static_cast<uint64>(seed_) << 32) | seed2_; } in seed()
1104 int seed2_; member in tensorflow::CudnnRNNKernelCommon
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | random_seed_ops.h | 36 seed2_((seed | seed2) == 0 ? random::New64() : seed2) {} in RandomSeeds()
41 int64 seed2() const { return seed2_; } in seed2()
47 const int64 seed2_; variable
|
| D | shuffle_dataset_op_test.cc | 36 seed2_(seed2), in ShuffleDatasetParams()
49 CreateTensor<int64>(TensorShape({}), {seed2_})}; in GetInputTensors()
92 int64 seed2_; member in tensorflow::data::__anonff9353e30111::ShuffleDatasetParams
|
| D | shuffle_dataset_op.cc | 158 seed_generator_->GenerateSeeds(&seed_, &seed2_); in Initialize()
241 seed_generator_->GenerateSeeds(&seed_, &seed2_); in GetNextInternal()
274 parent_generator_ = random::PhiloxRandom(seed_, seed2_); in ResetRngs()
290 TF_RETURN_IF_ERROR(writer->WriteScalar(this->full_name(kSeed2), seed2_)); in SaveInternal()
337 TF_RETURN_IF_ERROR(reader->ReadScalar(this->full_name(kSeed2), &seed2_)); in RestoreInternal()
414 int64 seed2_ TF_GUARDED_BY(mu_) = 0;
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | random_dataset_op_test.cc | 53 seed2_(CreateTensor<int64>(TensorShape({}), {seed2})) {}
56 return {seed_, seed2_}; in GetInputTensors()
77 Tensor seed2_; member in tensorflow::data::experimental::__anone47fe0d40111::RandomDatasetParams
|
| D | snapshot_dataset_op.cc | 971 OP_REQUIRES_OK(ctx, ctx->GetAttr("seed2", &seed2_)); in SnapshotDatasetOp()
1054 writer_buffer_size_, shuffle_on_read_, seed_, seed2_, in MakeDataset()
1085 seed2_(seed2), in Dataset()
1166 b->BuildAttrValue<int64>(seed2_, &seed2_attr); in AsGraphDefInternal()
1377 uint64 seed = dataset()->seed_ + dataset()->seed2_; in Initialize()
1378 if (dataset()->seed_ == 0 && dataset()->seed2_ == 0) { in Initialize()
2363 const uint64 seed2_; member in tensorflow::data::experimental::__anon94f7b5240311::SnapshotDatasetOp::Dataset
2399 int64 seed2_; member in tensorflow::data::experimental::__anon94f7b5240311::SnapshotDatasetOp
|