Home
last modified time | relevance | path

Searched refs:end_of_splits (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dsplit_utils_test.cc43 bool end_of_splits = false; in CheckOutput() local
44 while (!end_of_splits) { in CheckOutput()
46 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in CheckOutput()
47 if (!end_of_splits) { in CheckOutput()
80 bool end_of_splits = true; in TEST() local
81 TF_ASSERT_OK(split_provider.GetNext(&split, &end_of_splits)); in TEST()
82 EXPECT_FALSE(end_of_splits); in TEST()
87 bool end_of_splits = false; in TEST() local
88 TF_ASSERT_OK(split_provider.GetNext(&split, &end_of_splits)); in TEST()
89 EXPECT_TRUE(end_of_splits); in TEST()
[all …]
Dsplit_utils.cc28 Status IndexSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext() argument
31 *end_of_splits = true; in GetNext()
34 *end_of_splits = false; in GetNext()
68 Status ShardingSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext() argument
71 TF_RETURN_IF_ERROR(split_provider_->GetNext(split, end_of_splits)); in GetNext()
72 if (*end_of_splits) { in GetNext()
78 TF_RETURN_IF_ERROR(split_provider_->GetNext(split, end_of_splits)); in GetNext()
Dsplit_utils.h29 Status GetNext(Tensor* split, bool* end_of_splits) override;
49 Status GetNext(Tensor* split, bool* end_of_splits) override;
Drange_dataset_op.cc94 Status GetNext(Tensor* split, bool* end_of_splits) override { in GetNext() argument
95 int64 next = counter_.GetNext(end_of_splits); in GetNext()
96 if (*end_of_splits) { in GetNext()
/external/tensorflow/tensorflow/core/data/service/
Dsplit_provider.cc25 Status DataServiceSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext() argument
32 [this, split, end_of_splits] { in GetNext()
34 *end_of_splits); in GetNext()
Ddata_service.cc121 bool& end_of_splits) { in GetSplit() argument
132 end_of_splits = resp.end_of_splits(); in GetSplit()
133 if (!end_of_splits) { in GetSplit()
Dsplit_provider.h38 Status GetNext(Tensor* split, bool* end_of_splits) override;
Ddispatcher_impl.cc334 bool end_of_splits = false; in GetSplit() local
335 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in GetSplit()
336 TF_RETURN_IF_ERROR(RecordSplitProduced(job_id, repetition, end_of_splits)); in GetSplit()
337 response->set_end_of_splits(end_of_splits); in GetSplit()
338 if (end_of_splits) { in GetSplit()
345 VLOG(3) << "Returning from GetSplit, end_of_splits=" << end_of_splits; in GetSplit()
Ddata_service.h106 bool& end_of_splits);
Ddispatcher.proto48 bool end_of_splits = 2; field
/external/tensorflow/tensorflow/core/framework/
Ddataset.h316 virtual Status GetNext(Tensor* split, bool* end_of_splits) = 0;