Searched refs:split_provider (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/data/ |
| D | split_utils_test.cc | 33 Status SaveAndRestore(SplitProvider* split_provider) { in SaveAndRestore() argument
35 TF_RETURN_IF_ERROR(split_provider->Save(full_name, &writer)); in SaveAndRestore()
39 TF_RETURN_IF_ERROR(split_provider->Restore(full_name, &reader)); in SaveAndRestore()
43 Status CheckOutput(SplitProvider* split_provider, in CheckOutput() argument
49 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in CheckOutput()
59 IndexSplitProvider split_provider(0); in TEST() local
60 TF_EXPECT_OK(CheckOutput(&split_provider, in TEST()
65 IndexSplitProvider split_provider(1); in TEST() local
66 TF_EXPECT_OK(CheckOutput(&split_provider, in TEST()
71 IndexSplitProvider split_provider(3); in TEST() local
[all …]
|
| D | split_utils.cc | 68 std::shared_ptr<SplitProvider> split_provider) in ShardingSplitProvider() argument
71 split_provider_(split_provider), in ShardingSplitProvider()
|
| D | split_utils.h | 50 std::shared_ptr<SplitProvider> split_provider);
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | BUILD | 661 name = "split_provider",
662 srcs = ["split_provider.cc"],
664 "split_provider.h",
999 ":split_provider",
|
| D | dispatcher_impl.cc | 434 SplitProvider* split_provider = in GetSplit() local
436 DCHECK(split_provider != nullptr); in GetSplit()
439 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in GetSplit()
|