Searched refs:end_of_splits (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | split_utils_test.cc | 43 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 …]
|
D | split_utils.cc | 28 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()
|
D | split_utils.h | 29 Status GetNext(Tensor* split, bool* end_of_splits) override; 49 Status GetNext(Tensor* split, bool* end_of_splits) override;
|
D | range_dataset_op.cc | 94 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/ |
D | split_provider.cc | 25 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()
|
D | data_service.cc | 121 bool& end_of_splits) { in GetSplit() argument 132 end_of_splits = resp.end_of_splits(); in GetSplit() 133 if (!end_of_splits) { in GetSplit()
|
D | split_provider.h | 38 Status GetNext(Tensor* split, bool* end_of_splits) override;
|
D | dispatcher_impl.cc | 334 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()
|
D | data_service.h | 106 bool& end_of_splits);
|
D | dispatcher.proto | 48 bool end_of_splits = 2; field
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 316 virtual Status GetNext(Tensor* split, bool* end_of_splits) = 0;
|