Searched refs:end_of_splits (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/data/ |
| D | split_utils_test.cc | 46 bool end_of_splits = false; in CheckOutput() local
47 while (!end_of_splits) { in CheckOutput()
49 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in CheckOutput()
50 if (!end_of_splits) { in CheckOutput()
84 bool end_of_splits = true; in TEST() local
85 TF_ASSERT_OK(split_provider.GetNext(&split, &end_of_splits)); in TEST()
86 EXPECT_FALSE(end_of_splits); in TEST()
91 bool end_of_splits = false; in TEST() local
92 TF_ASSERT_OK(split_provider.GetNext(&split, &end_of_splits)); in TEST()
93 EXPECT_TRUE(end_of_splits); in TEST()
[all …]
|
| D | split_utils.cc | 34 Status IndexSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext() argument
37 *end_of_splits = true; in GetNext()
40 *end_of_splits = false; in GetNext()
74 Status ShardingSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext() argument
77 TF_RETURN_IF_ERROR(split_provider_->GetNext(split, end_of_splits)); in GetNext()
78 if (*end_of_splits) { in GetNext()
84 TF_RETURN_IF_ERROR(split_provider_->GetNext(split, end_of_splits)); in GetNext()
|
| D | split_utils.h | 32 Status GetNext(Tensor* split, bool* end_of_splits) override;
52 Status GetNext(Tensor* split, bool* end_of_splits) override;
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | split_provider.cc | 33 Status DataServiceSplitProvider::GetNext(Tensor* split, bool* end_of_splits) { in GetNext() argument
40 [this, split, end_of_splits] { in GetNext()
43 *end_of_splits); in GetNext()
48 if (*end_of_splits) { in GetNext()
|
| D | dispatcher_client.cc | 96 bool& end_of_splits) { in GetSplit() argument
108 end_of_splits = resp.end_of_splits(); in GetSplit()
109 if (!end_of_splits) { in GetSplit()
|
| D | split_provider.h | 44 Status GetNext(Tensor* split, bool* end_of_splits) override;
|
| D | dispatcher_client.h | 66 bool& end_of_splits);
|
| D | dispatcher_impl.cc | 438 bool end_of_splits = false; in GetSplit() local
439 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in GetSplit()
442 end_of_splits)); in GetSplit()
443 response->set_end_of_splits(end_of_splits); in GetSplit()
444 if (end_of_splits) { in GetSplit()
450 VLOG(3) << "Returning from GetSplit, end_of_splits=" << end_of_splits; in GetSplit()
|
| D | dispatcher.proto | 62 bool end_of_splits = 2; field
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op.cc | 137 bool end_of_splits; in GetNextInternal() local
139 TF_RETURN_IF_ERROR(split_provider_->GetNext(&split, &end_of_splits)); in GetNextInternal()
140 if (end_of_splits) { in GetNextInternal()
|
| D | range_dataset_op.cc | 120 Status GetNext(Tensor* split, bool* end_of_splits) override { in GetNext() argument
121 int64_t next = counter_.GetNext(end_of_splits); in GetNext()
122 if (*end_of_splits) { in GetNext()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.h | 367 virtual Status GetNext(Tensor* split, bool* end_of_splits) = 0;
|