| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | split_utils_test.cc | 30 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 …]
|
| D | repeat_dataset_op.cc | 166 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()
|
| D | tensor_slice_dataset_op.cc | 62 std::unique_ptr<SplitProvider>* split_provider) const override { in MakeSplitProvider()
63 *split_provider = in MakeSplitProvider()
117 split_provider_ = ctx->split_provider(); in Initialize()
|
| D | split_utils.cc | 62 std::shared_ptr<SplitProvider> split_provider) in ShardingSplitProvider() argument
65 split_provider_(split_provider), in ShardingSplitProvider()
|
| D | dataset_test_base.cc | 635 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()
|
| D | range_dataset_op.cc | 169 std::unique_ptr<SplitProvider>* split_provider) const override { in MakeSplitProvider()
170 *split_provider = in MakeSplitProvider()
203 split_provider_ = ctx->split_provider(); in Initialize()
|
| D | split_utils.h | 47 std::shared_ptr<SplitProvider> split_provider);
|
| D | shuffle_dataset_op.cc | 190 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()
|
| D | captured_function.cc | 480 if (ctx->split_provider() == nullptr) { in MakeIteratorFromInputElement()
486 params.split_provider = nullptr; in MakeIteratorFromInputElement()
|
| D | dataset_test_base.h | 544 std::unique_ptr<SplitProvider> split_provider,
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 101 Status Dataset::MakeIterator(std::unique_ptr<SplitProvider> split_provider, in MakeIterator() argument
119 params.split_provider = std::move(split_provider); in MakeIterator()
|
| D | standalone.h | 102 Status MakeIterator(std::unique_ptr<SplitProvider> split_provider,
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | dispatcher_impl.cc | 193 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()
|
| D | BUILD | 401 name = "split_provider",
402 srcs = ["split_provider.cc"],
404 "split_provider.h",
537 ":split_provider",
|
| D | worker_impl.cc | 156 auto split_provider = absl::make_unique<DataServiceSplitProvider>( in EnsureTaskInitialized() local
160 dataset->MakeIterator(std::move(split_provider), &iterator)); in EnsureTaskInitialized()
|
| D | dispatcher_impl.h | 160 std::unique_ptr<SplitProvider>& split_provider)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.h | 351 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;
|
| D | dataset.cc | 440 std::unique_ptr<SplitProvider>* split_provider) const { in MakeSplitProvider()
455 return inputs[0]->MakeSplitProvider(split_provider); in MakeSplitProvider()
|