Home
last modified time | relevance | path

Searched refs:MakeSplitProviders (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/data/
Dstandalone.cc160 Status Dataset::MakeSplitProviders( in MakeSplitProviders() function in tensorflow::data::standalone::Dataset
162 return finalized_dataset_->MakeSplitProviders(result); in MakeSplitProviders()
Dsplit_utils.cc138 TF_RETURN_IF_ERROR(input->MakeSplitProviders(&providers)); in GetSplitProviders()
203 TF_RETURN_IF_ERROR(input->MakeSplitProviders(&providers)); in GetSplitProviders()
Dstandalone.h109 Status MakeSplitProviders(
Ddataset_test_base.cc641 TF_RETURN_IF_ERROR(dataset->dataset()->MakeSplitProviders(&split_providers)); in CheckSplitProviderFullIteration()
656 TF_RETURN_IF_ERROR(dataset->dataset()->MakeSplitProviders(&split_providers)); in CheckSplitProviderShardedIteration()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drandom_dataset_op.cc54 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::experimental::RandomDatasetOp::Dataset
Dlist_dataset_op.cc63 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::ListDatasetOp::Dataset
Ddirected_interleave_dataset_op.cc82 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::experimental::DirectedInterleaveDatasetOp::Dataset
Dsnapshot_dataset_op.cc145 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::experimental::SnapshotDatasetV2Op::Dataset
996 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::experimental::__anonbc65ddc20311::SnapshotDatasetOp::Dataset
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_dataset_op.cc59 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::TensorDatasetOp::Dataset
Dconcatenate_dataset_op.cc70 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::ConcatenateDatasetOp::Dataset
Dzip_dataset_op.cc70 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::ZipDatasetOp::Dataset
Dtensor_slice_dataset_op.cc70 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::TensorSliceDatasetOp::Dataset
Drange_dataset_op.cc211 Status MakeSplitProviders(std::vector<std::unique_ptr<SplitProvider>>* in MakeSplitProviders() function in tensorflow::data::RangeDatasetOp::Dataset
/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_impl.cc267 MakeSplitProviders(iteration.job->dataset_id, split_providers)); in RestoreSplitProviders()
454 Status DataServiceDispatcherImpl::MakeSplitProviders( in MakeSplitProviders() function in tensorflow::data::DataServiceDispatcherImpl
466 TF_RETURN_IF_ERROR(standalone_dataset->MakeSplitProviders(&split_providers)); in MakeSplitProviders()
764 MakeSplitProviders(job->dataset_id, split_providers_[iteration_id])); in CreateIteration()
Ddispatcher_impl.h191 Status MakeSplitProviders(
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc709 Status DatasetBase::MakeSplitProviders( in MakeSplitProviders() function in tensorflow::data::DatasetBase
726 return inputs[0]->MakeSplitProviders(split_providers); in MakeSplitProviders()
Ddataset.h967 virtual Status MakeSplitProviders(