Home
last modified time | relevance | path

Searched refs:DeterminismPolicy (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_interleave_dataset_op_test.cc148 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelInterleaveDatasetParams1()
173 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelInterleaveDatasetParams2()
198 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams3()
223 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams4()
248 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams5()
273 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams6()
298 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelInterleaveDatasetParams7()
323 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams8()
348 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams9()
373 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams10()
[all …]
Dparallel_map_dataset_op_test.cc121 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelMapDatasetParams1()
139 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelMapDatasetParams2()
157 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelMapDatasetParams3()
175 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelMapDatasetParams4()
193 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelMapDatasetParams5()
211 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelMapDatasetParams6()
230 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelMapDatasetParams7()
250 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelMapDatasetParams8()
266 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelMapDatasetParamsWithInvalidNumParallelCalls()
Ddataset_utils.cc485 Status DeterminismPolicy::FromString(const std::string& s, in FromString()
486 DeterminismPolicy* out) { in FromString()
487 DeterminismPolicy::Type type; in FromString()
488 if (s == DeterminismPolicy::kDeterministic) { in FromString()
489 type = DeterminismPolicy::Type::kDeterministic; in FromString()
490 } else if (s == DeterminismPolicy::kNondeterministic) { in FromString()
491 type = DeterminismPolicy::Type::kNondeterministic; in FromString()
492 } else if (s == DeterminismPolicy::kDefault) { in FromString()
493 type = DeterminismPolicy::Type::kDefault; in FromString()
497 *out = DeterminismPolicy(type); in FromString()
[all …]
Ddataset_utils_test.cc274 DeterminismPolicy determinism; in TEST()
275 TF_ASSERT_OK(DeterminismPolicy::FromString("true", &determinism)); in TEST()
277 TF_ASSERT_OK(DeterminismPolicy::FromString("false", &determinism)); in TEST()
279 TF_ASSERT_OK(DeterminismPolicy::FromString("default", &determinism)); in TEST()
285 DeterminismPolicy determinism; in TEST()
286 TF_ASSERT_OK(DeterminismPolicy::FromString(s, &determinism)); in TEST()
292 EXPECT_TRUE(DeterminismPolicy(true).IsDeterministic()); in TEST()
293 EXPECT_FALSE(DeterminismPolicy(true).IsNondeterministic()); in TEST()
294 EXPECT_FALSE(DeterminismPolicy(true).IsDefault()); in TEST()
296 EXPECT_TRUE(DeterminismPolicy(false).IsNondeterministic()); in TEST()
[all …]
Ddataset_utils.h116 class DeterminismPolicy {
132 DeterminismPolicy() : determinism_(Type::kDefault) {} in DeterminismPolicy() function
133 explicit DeterminismPolicy(Type determinism) : determinism_(determinism) {} in DeterminismPolicy() function
136 explicit DeterminismPolicy(bool is_deterministic);
138 static Status FromString(const std::string& s, DeterminismPolicy* out);
Dparallel_map_dataset_op.cc77 DeterminismPolicy deterministic, in Dataset()
669 const DeterminismPolicy deterministic_;
689 DeterminismPolicy(DeterminismPolicy::Type::kNondeterministic); in ParallelMapDatasetOp()
691 deterministic_ = DeterminismPolicy(DeterminismPolicy::Type::kDefault); in ParallelMapDatasetOp()
698 ctx, DeterminismPolicy::FromString(deterministic, &deterministic_)); in ParallelMapDatasetOp()
Dparallel_map_dataset_op.h56 DeterminismPolicy deterministic_;
Dparallel_interleave_dataset_op.h56 DeterminismPolicy deterministic_;
Dparallel_interleave_dataset_op.cc165 DeterminismPolicy deterministic, const DataTypeVector& output_types, in Dataset()
1563 const DeterminismPolicy deterministic_;
1583 DeterminismPolicy(DeterminismPolicy::Type::kNondeterministic); in ParallelInterleaveDatasetOp()
1585 deterministic_ = DeterminismPolicy(DeterminismPolicy::Type::kDefault); in ParallelInterleaveDatasetOp()
1592 ctx, DeterminismPolicy::FromString(deterministic, &deterministic_)); in ParallelInterleaveDatasetOp()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparallel_interleave_dataset_op_test.cc135 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelInterleaveDatasetParams1()
161 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelInterleaveDatasetParams2()
187 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams3()
213 /*deterministic=*/DeterminismPolicy::kNondeterministic, in ParallelInterleaveDatasetParams4()
239 /*deterministic=*/DeterminismPolicy::kDeterministic, in ParallelInterleaveDatasetParams5()
262 /*deterministic=*/DeterminismPolicy::kNondeterministic, in EmptyInputParams()
286 /*deterministic=*/DeterminismPolicy::kDeterministic, in InvalidCycleLengthParams()
310 /*deterministic=*/DeterminismPolicy::kDeterministic, in InvalidBlockLengthParams()
334 /*deterministic=*/DeterminismPolicy::kDeterministic, in InvalidBufferOutputElementsParams()
358 /*deterministic=*/DeterminismPolicy::kDeterministic, in InvalidPrefetchInputElementsParams()
Dparse_example_dataset_op.cc66 DeterminismPolicy(DeterminismPolicy::Type::kNondeterministic); in ParseExampleDatasetOp()
68 deterministic_ = DeterminismPolicy(DeterminismPolicy::Type::kDefault); in ParseExampleDatasetOp()
75 ctx, DeterminismPolicy::FromString(deterministic, &deterministic_)); in ParseExampleDatasetOp()
222 const DeterminismPolicy& deterministic, bool has_ragged_keys, in Dataset()
923 const DeterminismPolicy deterministic_;
931 DeterminismPolicy deterministic_;
Dparallel_interleave_dataset_op.h60 DeterminismPolicy deterministic_;
Dparallel_interleave_dataset_op.cc97 int64 block_length, DeterminismPolicy deterministic, in Dataset()
1163 const DeterminismPolicy deterministic_;
1182 ctx, DeterminismPolicy::FromString(deterministic, &deterministic_)); in ParallelInterleaveDatasetOp()
1206 DeterminismPolicy(DeterminismPolicy::Type::kNondeterministic); in MakeDataset()
1209 DeterminismPolicy(DeterminismPolicy::Type::kDeterministic); in MakeDataset()