Home
last modified time | relevance | path

Searched refs:split_provider_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dsplit_utils.cc65 split_provider_(split_provider), in ShardingSplitProvider()
71 TF_RETURN_IF_ERROR(split_provider_->GetNext(split, end_of_splits)); in GetNext()
78 TF_RETURN_IF_ERROR(split_provider_->GetNext(split, end_of_splits)); in GetNext()
84 TF_RETURN_IF_ERROR(split_provider_->Reset()); in Reset()
93 TF_RETURN_IF_ERROR(split_provider_->Save( in Save()
105 TF_RETURN_IF_ERROR(split_provider_->Restore( in Restore()
Dtensor_slice_dataset_op.cc117 split_provider_ = ctx->split_provider(); in Initialize()
118 if (split_provider_ == nullptr) { in Initialize()
119 split_provider_ = std::make_shared<IndexSplitProvider>( in Initialize()
129 TF_RETURN_IF_ERROR(split_provider_->GetNext(&split, end_of_sequence)); in GetNextInternal()
155 return split_provider_->Save( in SaveInternal()
161 return split_provider_->Restore( in RestoreInternal()
166 std::shared_ptr<SplitProvider> split_provider_; member in tensorflow::data::TensorSliceDatasetOp::Dataset::Iterator
Drange_dataset_op.cc203 split_provider_ = ctx->split_provider(); in Initialize()
204 if (!split_provider_) { in Initialize()
215 if (split_provider_ != nullptr) { in GetNextInternal()
217 TF_RETURN_IF_ERROR(split_provider_->GetNext(&split, end_of_sequence)); in GetNextInternal()
253 if (split_provider_) { in SaveInternal()
256 TF_RETURN_IF_ERROR(split_provider_->Save( in SaveInternal()
271 TF_RETURN_IF_ERROR(split_provider_->Restore( in RestoreInternal()
290 std::shared_ptr<SplitProvider> split_provider_; member in tensorflow::data::RangeDatasetOp::Dataset::Iterator
Dsplit_utils.h60 std::shared_ptr<SplitProvider> split_provider_ TF_GUARDED_BY(mu_);