Home
last modified time | relevance | path

Searched refs:kSeed2 (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Drandom_seed_ops.cc32 const char kSeed2[] = "seed2"; variable
73 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64_t>(ctx, kSeed2, &seed2)); in Compute()
Dshuffle_dataset_op.h28 static constexpr const char* const kSeed2 = "seed2"; variable
Dshuffle_dataset_op.cc48 /* static */ constexpr const char* const ShuffleDatasetOpBase::kSeed2; member in tensorflow::data::ShuffleDatasetOpBase
263 TF_RETURN_IF_ERROR(writer->WriteScalar(this->full_name(kSeed2), seed2_)); in SaveInternal()
310 TF_RETURN_IF_ERROR(reader->ReadScalar(this->full_name(kSeed2), &seed2_)); in RestoreInternal()
731 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64_t>(ctx, kSeed2, &seed2)); in MakeDataset()
797 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64_t>(ctx, kSeed2, &seed2)); in MakeDataset()
972 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64_t>(ctx, kSeed2, &seed2)); in MakeDataset()
Dshuffle_dataset_op_test.cc67 input_names->emplace_back(ShuffleDatasetOpBase::kSeed2); in GetInputNames()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drandom_dataset_op.h33 static constexpr const char* const kSeed2 = "seed2"; variable
Dsampling_dataset_op.h35 static constexpr const char* const kSeed2 = "seed2"; variable
Dsampling_dataset_op.cc37 /* static */ constexpr const char* const SamplingDatasetOp::kSeed2; member in tensorflow::data::experimental::SamplingDatasetOp
233 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64_t>(ctx, kSeed2, &seed2)); in MakeDataset()
Dsampling_dataset_op_test.cc51 SamplingDatasetOp::kSeed, SamplingDatasetOp::kSeed2}; in GetInputNames()
Drandom_dataset_op_test.cc62 *input_names = {RandomDatasetOp::kSeed, RandomDatasetOp::kSeed2}; in GetInputNames()
Drandom_dataset_op.cc39 /* static */ constexpr const char* const RandomDatasetOp::kSeed2; member in tensorflow::data::experimental::RandomDatasetOp