Searched refs:SplitProvider (Results 1 – 15 of 15) sorted by relevance
26 class IndexSplitProvider : public SplitProvider {44 class ShardingSplitProvider : public SplitProvider {47 std::shared_ptr<SplitProvider> split_provider);60 std::shared_ptr<SplitProvider> split_provider_ TF_GUARDED_BY(mu_);
30 Status SaveAndRestore(SplitProvider* split_provider) { in SaveAndRestore()40 Status CheckOutput(SplitProvider* split_provider, in CheckOutput()
89 class RangeDatasetOp::RangeSplitProvider : public SplitProvider {169 std::unique_ptr<SplitProvider>* split_provider) const override { in MakeSplitProvider()290 std::shared_ptr<SplitProvider> split_provider_;
62 std::unique_ptr<SplitProvider>* split_provider) const override { in MakeSplitProvider()166 std::shared_ptr<SplitProvider> split_provider_;
62 std::shared_ptr<SplitProvider> split_provider) in ShardingSplitProvider()
322 TEST_F(TensorSliceDatasetOpTest, SplitProvider) { in TEST_F() argument
635 std::unique_ptr<SplitProvider> split_provider; in CheckSplitProviderFullIteration()650 std::unique_ptr<SplitProvider> split_provider; in CheckSplitProviderShardedIteration()859 std::unique_ptr<SplitProvider> split_provider, in MakeIterator()
544 std::unique_ptr<SplitProvider> split_provider,
102 Status MakeIterator(std::unique_ptr<SplitProvider> split_provider,106 Status MakeSplitProvider(std::unique_ptr<SplitProvider>* result);
101 Status Dataset::MakeIterator(std::unique_ptr<SplitProvider> split_provider, in MakeIterator()138 Status Dataset::MakeSplitProvider(std::unique_ptr<SplitProvider>* result) { in MakeSplitProvider()
155 std::unique_ptr<SplitProvider>& restored)160 std::unique_ptr<SplitProvider>& split_provider)257 absl::flat_hash_map<int64, std::unique_ptr<SplitProvider>> split_providers_
28 class DataServiceSplitProvider : public SplitProvider {
188 const Job& job, std::unique_ptr<SplitProvider>& restored) in RestoreSplitProvider()193 std::unique_ptr<SplitProvider> split_provider; in RestoreSplitProvider()331 SplitProvider* split_provider = split_providers_[job_id].get(); in GetSplit()350 int64 dataset_id, std::unique_ptr<SplitProvider>& split_provider) in MakeSplitProvider()
311 class SplitProvider {313 virtual ~SplitProvider() {} in ~SplitProvider()420 std::shared_ptr<SplitProvider> split_provider = nullptr;468 std::shared_ptr<SplitProvider> split_provider() { in split_provider()867 std::unique_ptr<SplitProvider>* split_provider) const;
440 std::unique_ptr<SplitProvider>* split_provider) const { in MakeSplitProvider()