Searched defs:split_provider (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | split_utils_test.cc | 30 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 …]
|
D | split_utils.cc | 62 std::shared_ptr<SplitProvider> split_provider) in ShardingSplitProvider()
|
D | dataset_test_base.cc | 635 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/ |
D | standalone.cc | 101 Status Dataset::MakeIterator(std::unique_ptr<SplitProvider> split_provider, in MakeIterator()
|
/external/tensorflow/tensorflow/core/data/service/ |
D | dispatcher_impl.cc | 193 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()
|
D | worker_impl.cc | 156 auto split_provider = absl::make_unique<DataServiceSplitProvider>( in EnsureTaskInitialized() local
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 420 std::shared_ptr<SplitProvider> split_provider = nullptr; member 468 std::shared_ptr<SplitProvider> split_provider() { in split_provider() function
|