Home
last modified time | relevance | path

Searched refs:split_provider (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dsplit_utils_test.cc30 Status SaveAndRestore(SplitProvider* split_provider) { in SaveAndRestore() argument
32 TF_RETURN_IF_ERROR(split_provider->Save(full_name, &writer)); in SaveAndRestore()
36 TF_RETURN_IF_ERROR(split_provider->Restore(full_name, &reader)); in SaveAndRestore()
40 Status CheckOutput(SplitProvider* split_provider, in CheckOutput() argument
46 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in CheckOutput()
56 IndexSplitProvider split_provider(0); in TEST() local
58 CheckOutput(&split_provider, CreateTensors<int64>(TensorShape({}), {}))); in TEST()
62 IndexSplitProvider split_provider(1); in TEST() local
63 TF_EXPECT_OK(CheckOutput(&split_provider, in TEST()
68 IndexSplitProvider split_provider(3); in TEST() local
[all …]
Drepeat_dataset_op.cc166 if (ctx->split_provider()) { in GetNextInternal()
167 TF_RETURN_IF_ERROR(ctx->split_provider()->Reset()); in GetNextInternal()
237 if (first_call_ && *end_of_sequence && !ctx->split_provider()) { in GetNextInternal()
249 if (ctx->split_provider()) { in GetNextInternal()
250 TF_RETURN_IF_ERROR(ctx->split_provider()->Reset()); in GetNextInternal()
Dtensor_slice_dataset_op.cc62 std::unique_ptr<SplitProvider>* split_provider) const override { in MakeSplitProvider()
63 *split_provider = in MakeSplitProvider()
117 split_provider_ = ctx->split_provider(); in Initialize()
Dsplit_utils.cc62 std::shared_ptr<SplitProvider> split_provider) in ShardingSplitProvider() argument
65 split_provider_(split_provider), in ShardingSplitProvider()
Ddataset_test_base.cc635 std::unique_ptr<SplitProvider> split_provider; in CheckSplitProviderFullIteration() local
636 TF_RETURN_IF_ERROR(dataset->dataset()->MakeSplitProvider(&split_provider)); in CheckSplitProviderFullIteration()
639 MakeIterator(params, *dataset, std::move(split_provider), &iterator)); in CheckSplitProviderFullIteration()
650 std::unique_ptr<SplitProvider> split_provider; in CheckSplitProviderShardedIteration() local
651 TF_RETURN_IF_ERROR(dataset->dataset()->MakeSplitProvider(&split_provider)); in CheckSplitProviderShardedIteration()
652 split_provider = absl::make_unique<ShardingSplitProvider>( in CheckSplitProviderShardedIteration()
653 num_shards, shard_index, std::move(split_provider)); in CheckSplitProviderShardedIteration()
658 iterator_params.split_provider = std::move(split_provider); in CheckSplitProviderShardedIteration()
859 std::unique_ptr<SplitProvider> split_provider, in MakeIterator() argument
865 iterator_params.split_provider = std::move(split_provider); in MakeIterator()
Drange_dataset_op.cc169 std::unique_ptr<SplitProvider>* split_provider) const override { in MakeSplitProvider()
170 *split_provider = in MakeSplitProvider()
203 split_provider_ = ctx->split_provider(); in Initialize()
Dsplit_utils.h47 std::shared_ptr<SplitProvider> split_provider);
Dshuffle_dataset_op.cc190 if (ctx->split_provider() == nullptr && !data_produced_ && in GetNextInternal()
201 if (ctx->split_provider()) { in GetNextInternal()
202 TF_RETURN_IF_ERROR(ctx->split_provider()->Reset()); in GetNextInternal()
Dcaptured_function.cc480 if (ctx->split_provider() == nullptr) { in MakeIteratorFromInputElement()
486 params.split_provider = nullptr; in MakeIteratorFromInputElement()
Ddataset_test_base.h544 std::unique_ptr<SplitProvider> split_provider,
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc101 Status Dataset::MakeIterator(std::unique_ptr<SplitProvider> split_provider, in MakeIterator() argument
119 params.split_provider = std::move(split_provider); in MakeIterator()
Dstandalone.h102 Status MakeIterator(std::unique_ptr<SplitProvider> split_provider,
/external/tensorflow/tensorflow/core/data/service/
Ddispatcher_impl.cc193 std::unique_ptr<SplitProvider> split_provider; in RestoreSplitProvider() local
194 TF_RETURN_IF_ERROR(MakeSplitProvider(job.dataset_id, split_provider)); in RestoreSplitProvider()
199 split_provider->GetNext(&unused_tensor, &unused_end_of_splits)); in RestoreSplitProvider()
201 restored = std::move(split_provider); in RestoreSplitProvider()
331 SplitProvider* split_provider = split_providers_[job_id].get(); in GetSplit() local
332 DCHECK(split_provider != nullptr); in GetSplit()
335 TF_RETURN_IF_ERROR(split_provider->GetNext(&split, &end_of_splits)); in GetSplit()
350 int64 dataset_id, std::unique_ptr<SplitProvider>& split_provider) in MakeSplitProvider() argument
360 TF_RETURN_IF_ERROR(standalone_dataset->MakeSplitProvider(&split_provider)); in MakeSplitProvider()
DBUILD401 name = "split_provider",
402 srcs = ["split_provider.cc"],
404 "split_provider.h",
537 ":split_provider",
Dworker_impl.cc156 auto split_provider = absl::make_unique<DataServiceSplitProvider>( in EnsureTaskInitialized() local
160 dataset->MakeIterator(std::move(split_provider), &iterator)); in EnsureTaskInitialized()
Ddispatcher_impl.h160 std::unique_ptr<SplitProvider>& split_provider)
/external/tensorflow/tensorflow/core/framework/
Ddataset.h351 split_provider(ctx->split_provider()), in Params()
420 std::shared_ptr<SplitProvider> split_provider = nullptr; member
468 std::shared_ptr<SplitProvider> split_provider() { in split_provider() function
469 return params_.split_provider; in split_provider()
867 std::unique_ptr<SplitProvider>* split_provider) const;
Ddataset.cc440 std::unique_ptr<SplitProvider>* split_provider) const { in MakeSplitProvider()
455 return inputs[0]->MakeSplitProvider(split_provider); in MakeSplitProvider()