Searched refs:MakeSplitProvider (Results 1 – 9 of 9) sorted by relevance
138 Status Dataset::MakeSplitProvider(std::unique_ptr<SplitProvider>* result) { in MakeSplitProvider() function in tensorflow::data::standalone::Dataset139 return dataset_->MakeSplitProvider(result); in MakeSplitProvider()
106 Status MakeSplitProvider(std::unique_ptr<SplitProvider>* result);
194 TF_RETURN_IF_ERROR(MakeSplitProvider(job.dataset_id, split_provider)); in RestoreSplitProvider()341 MakeSplitProvider(job->dataset_id, split_providers_[job_id])); in GetSplit()349 Status DataServiceDispatcherImpl::MakeSplitProvider( in MakeSplitProvider() function in tensorflow::data::DataServiceDispatcherImpl360 TF_RETURN_IF_ERROR(standalone_dataset->MakeSplitProvider(&split_provider)); in MakeSplitProvider()521 TF_RETURN_IF_ERROR(MakeSplitProvider(dataset_id, split_providers_[job_id])); in CreateJob()
159 Status MakeSplitProvider(int64 dataset_id,
61 Status MakeSplitProvider( in MakeSplitProvider() function in tensorflow::data::TensorSliceDatasetOp::Dataset
168 Status MakeSplitProvider( in MakeSplitProvider() function in tensorflow::data::RangeDatasetOp::Dataset
636 TF_RETURN_IF_ERROR(dataset->dataset()->MakeSplitProvider(&split_provider)); in CheckSplitProviderFullIteration()651 TF_RETURN_IF_ERROR(dataset->dataset()->MakeSplitProvider(&split_provider)); in CheckSplitProviderShardedIteration()
439 Status DatasetBase::MakeSplitProvider( in MakeSplitProvider() function in tensorflow::data::DatasetBase455 return inputs[0]->MakeSplitProvider(split_provider); in MakeSplitProvider()
866 virtual Status MakeSplitProvider(