Home
last modified time | relevance | path

Searched defs:split_provider (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dsplit_utils_test.cc30 Status SaveAndRestore(SplitProvider* split_provider) { in SaveAndRestore()
40 Status CheckOutput(SplitProvider* split_provider, in CheckOutput()
56 IndexSplitProvider split_provider(0); in TEST() local
62 IndexSplitProvider split_provider(1); in TEST() local
68 IndexSplitProvider split_provider(3); in TEST() local
74 IndexSplitProvider split_provider(4); in TEST() local
94 ShardingSplitProvider split_provider(2, 0, base); in TEST() local
101 ShardingSplitProvider split_provider(2, 1, base); in TEST() local
108 ShardingSplitProvider split_provider(3, 1, base); in TEST() local
115 ShardingSplitProvider split_provider(2, 1, base); in TEST() local
[all …]
Dsplit_utils.cc62 std::shared_ptr<SplitProvider> split_provider) in ShardingSplitProvider()
Ddataset_test_base.cc635 std::unique_ptr<SplitProvider> split_provider; in CheckSplitProviderFullIteration() local
650 std::unique_ptr<SplitProvider> split_provider; in CheckSplitProviderShardedIteration() local
859 std::unique_ptr<SplitProvider> split_provider, in MakeIterator()
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc101 Status Dataset::MakeIterator(std::unique_ptr<SplitProvider> split_provider, in MakeIterator()
/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_impl.cc193 std::unique_ptr<SplitProvider> split_provider; in RestoreSplitProvider() local
331 SplitProvider* split_provider = split_providers_[job_id].get(); in GetSplit() local
350 int64 dataset_id, std::unique_ptr<SplitProvider>& split_provider) in MakeSplitProvider()
Dworker_impl.cc156 auto split_provider = absl::make_unique<DataServiceSplitProvider>( in EnsureTaskInitialized() local
/external/tensorflow/tensorflow/core/framework/
Ddataset.h420 std::shared_ptr<SplitProvider> split_provider = nullptr; member
468 std::shared_ptr<SplitProvider> split_provider() { in split_provider() function