| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 126 TF_RETURN_IF_ERROR(dataset_->MakeIterator(ctx.get(), /*parent=*/nullptr, in MakeIterator()
139 return dataset_->MakeSplitProvider(result); in MakeSplitProvider()
142 const DatasetBase* Dataset::Get() const { return dataset_; } in Get()
147 : dataset_(dataset), in Dataset()
157 Dataset::~Dataset() { dataset_->Unref(); } in ~Dataset()
|
| D | standalone.h | 115 DatasetBase* dataset_; // owned variable
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 56 DatasetVariantWrapper() : dataset_(nullptr) {} in DatasetVariantWrapper()
59 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {} in DatasetVariantWrapper()
62 : dataset_(other.dataset_) { in DatasetVariantWrapper()
63 if (dataset_) dataset_->Ref(); in DatasetVariantWrapper()
68 std::swap(dataset_, other.dataset_); in operator =()
75 if (dataset_) dataset_->Unref(); in ~DatasetVariantWrapper()
78 DatasetBase* get() const { return dataset_; } in get()
82 if (dataset_) { in DebugString()
83 return dataset_->DebugString(); in DebugString()
99 DatasetBase* dataset_; // Owns one reference. member in tensorflow::data::__anon260a7bcb0111::DatasetVariantWrapper
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | repeat_dataset_op_test.cc | 141 if (dataset_->Cardinality() == kInfiniteCardinality) { in TEST_P()
316 bool end_of_sequence = dataset_->Cardinality() == 0; in TEST_P()
328 *dataset_, &iterator_)); in TEST_P()
342 if (dataset_->Cardinality() == kInfiniteCardinality && in TEST_P()
348 if (breakpoint >= dataset_->Cardinality()) { in TEST_P()
349 if (dataset_->Cardinality() == kInfiniteCardinality) { in TEST_P()
|
| D | cache_dataset_ops_test.cc | 184 TF_ASSERT_OK(dataset_->MakeIterator( in TEST_P()
324 TF_ASSERT_OK(dataset_->MakeIterator( in TEST_P()
341 *dataset_, &iterator_)); in TEST_P()
355 if (breakpoint >= dataset_->Cardinality()) { in TEST_P()
|
| D | dataset_test_base.cc | 211 if (dataset_) { in ~DatasetOpsTestBase()
212 dataset_->Unref(); in ~DatasetOpsTestBase()
674 EXPECT_EQ(dataset_->node_name(), expected_dataset_node_name); in CheckDatasetNodeName()
680 EXPECT_EQ(dataset_->type_string(), expected_type_str); in CheckDatasetTypeString()
687 VerifyTypesMatch(dataset_->output_dtypes(), expected_output_dtypes)); in CheckDatasetOutputDtypes()
693 TF_EXPECT_OK(VerifyShapesCompatible(dataset_->output_shapes(), in CheckDatasetOutputShapes()
699 EXPECT_EQ(dataset_->Cardinality(), expected_cardinality); in CheckDatasetCardinality()
762 return CheckIteratorSaveAndRestore(dataset_, iterator_ctx_.get(), in CheckIteratorSaveAndRestore()
775 &dataset_ctx_, &tensors_, &dataset_)); in Initialize()
778 dataset_->MakeIterator(iterator_ctx_.get(), /*parent=*/nullptr, in Initialize()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 376 int64 num_slices = dataset_->Cardinality(); in TEST_P()
408 *dataset_, &iterator_)); in TEST_P()
|
| D | shuffle_dataset_op_test.cc | 344 TF_ASSERT_OK(dataset_->MakeIterator( in TEST_P()
533 *dataset_, &iterator_)); in TEST_P()
|
| D | dataset_test_base.h | 487 dataset_(dataset), in TestDataset()
490 DatasetBase* dataset() const { return dataset_; } in dataset()
501 DatasetBase* dataset_; variable
781 DatasetBase* dataset_ = nullptr; variable
|
| D | tensor_dataset_op_test.cc | 245 *dataset_, &iterator_)); in TEST_P()
|
| D | tensor_slice_dataset_op_test.cc | 314 *dataset_, &iterator_)); in TEST_P()
|
| D | window_dataset_op_test.cc | 483 *dataset_, &iterator_)); in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 403 explicit DatasetIterator(data::DatasetBase* dataset) : dataset_(dataset) {} in DatasetIterator()
417 TF_RETURN_IF_ERROR(dataset_->MakeIterator(iterator_ctx_.get(), nullptr, in Init()
441 int64 size = dataset_->Cardinality(); in total_size()
449 data::DatasetBase* dataset_; // not owned. member in tensorflow::lookup::DatasetIterator
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | task_runner.cc | 74 : dataset_(std::move(dataset)), iterator_(std::move(iterator)) {} in StandaloneTaskIterator()
82 return dataset_->Get()->Cardinality(); in Cardinality()
|
| D | task_runner.h | 51 std::unique_ptr<standalone::Dataset> dataset_;
|