Home
last modified time | relevance | path

Searched refs:dataset_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dindexed_dataset_op.cc164 IndexedDatasetVariantWrapper() : dataset_(nullptr) {} in IndexedDatasetVariantWrapper()
168 : dataset_(dataset) {} in IndexedDatasetVariantWrapper()
171 : dataset_(other.dataset_) { in IndexedDatasetVariantWrapper()
172 if (dataset_) dataset_->Ref(); in IndexedDatasetVariantWrapper()
176 if (dataset_) dataset_->Unref(); in ~IndexedDatasetVariantWrapper()
179 IndexedDataset* get() const { return dataset_; } in get()
183 if (dataset_) { in DebugString()
184 return dataset_->DebugString(); in DebugString()
202 IndexedDataset* const dataset_; // Owns one reference. member in tensorflow::data::__anon0a1add170111::IndexedDatasetVariantWrapper
492 explicit Materialized(Dataset* dataset) : dataset_(dataset) { in Materialized()
[all …]
Dparse_example_dataset_op.cc263 : dataset_(dataset) {} in ParseExampleFunctor()
279 example::FastParseExampleConfig config = dataset_->config_; in MapFunc()
289 (*output).resize(dataset_->key_to_output_index_.size()); in MapFunc()
290 for (int d = 0; d < dataset_->dense_keys_.size(); ++d) { in MapFunc()
292 dataset_->key_to_output_index_.at(dataset_->dense_keys_[d]); in MapFunc()
294 dataset_->output_dtypes()[output_index]) in MapFunc()
297 << DataTypeString(dataset_->output_dtypes()[output_index]) in MapFunc()
301 DCHECK(dataset_->output_shapes()[output_index].IsCompatibleWith( in MapFunc()
305 << dataset_->output_shapes()[output_index].DebugString() in MapFunc()
311 for (int d = 0; d < dataset_->sparse_keys_.size(); ++d) { in MapFunc()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc209 : dataset_(dataset) {} in ShortCircuitFunctor()
215 dataset_->captured_func_->captured_inputs(); in MapFunc()
217 for (size_t i = 0; i < dataset_->short_circuit_indices_.size(); ++i) { in MapFunc()
218 if (dataset_->short_circuit_indices_[i] < num_args) { in MapFunc()
219 if (dataset_->can_move_[i]) { in MapFunc()
221 input_element[dataset_->short_circuit_indices_[i]])); in MapFunc()
224 input_element[dataset_->short_circuit_indices_[i]]); in MapFunc()
228 captured_inputs[dataset_->short_circuit_indices_[i] - in MapFunc()
235 const Dataset* const dataset_; member in tensorflow::data::__anon58e4a4830111::ParallelMapDatasetOp::Dataset::ShortCircuitFunctor
241 : dataset_(dataset) {} in ParallelMapDatasetFunctor()
[all …]
Dconcatenate_dataset_op_test.cc137 if (dataset_) dataset_->Unref(); in ~ConcatenateDatasetOpTestHelper()
155 dataset_kernel_ctx_.get(), &dataset_)); in CreateDatasetFromTestCase()
164 dataset_->MakeIterator(iterator_ctx_.get(), "Iterator", &iterator_)); in CreateIteratorFromTestCase()
170 DatasetBase *dataset_ = nullptr; // owned by this class. member in tensorflow::data::__anonda562e9d0111::ConcatenateDatasetOpTestHelper
228 EXPECT_EQ(dataset_->type_string(), kOpName); in TEST_F()
237 TF_EXPECT_OK(VerifyTypesMatch(dataset_->output_dtypes(), in TEST_P()
247 TF_EXPECT_OK(VerifyShapesCompatible(dataset_->output_shapes(), in TEST_P()
258 EXPECT_EQ(dataset_->Cardinality(), GetParam().expected_cardinality); in TEST_P()
271 TF_ASSERT_OK(dataset_->Save(serialization_ctx.get(), &writer)); in TEST_F()
339 if (breakpoint >= dataset_->Cardinality()) { in TEST_P()
Dzip_dataset_op_test.cc119 if (dataset_) dataset_->Unref(); in ~ZipDatasetOpTestHelper()
140 dataset_kernel_ctx_.get(), &dataset_)); in CreateDatasetFromTestCase()
149 dataset_->MakeIterator(iterator_ctx_.get(), "Iterator", &iterator_)); in CreateIteratorFromTestCase()
155 DatasetBase *dataset_ = nullptr; // owned by this class. member in tensorflow::data::__anon63a5714b0111::ZipDatasetOpTestHelper
194 EXPECT_EQ(dataset_->type_string(), kOpName); in TEST_F()
212 VerifyTypesMatch(dataset_->output_dtypes(), expected_output_dtypes)); in TEST_P()
229 TF_EXPECT_OK(VerifyShapesCompatible(dataset_->output_shapes(), in TEST_P()
241 EXPECT_EQ(dataset_->Cardinality(), in TEST_P()
255 TF_ASSERT_OK(dataset_->Save(serialization_ctx.get(), &writer)); in TEST_F()
338 if (breakpoint >= dataset_->Cardinality()) { in TEST_P()
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc41 DatasetVariantWrapper() : dataset_(nullptr) {} in DatasetVariantWrapper()
44 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {} in DatasetVariantWrapper()
47 : dataset_(other.dataset_) { in DatasetVariantWrapper()
48 if (dataset_) dataset_->Ref(); in DatasetVariantWrapper()
52 if (dataset_) dataset_->Unref(); in ~DatasetVariantWrapper()
55 DatasetBase* get() const { return dataset_; } in get()
59 if (dataset_) { in DebugString()
60 return dataset_->DebugString(); in DebugString()
76 DatasetBase* const dataset_; // Owns one reference. member in tensorflow::data::__anon86842c2d0111::DatasetVariantWrapper
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc104 TF_RETURN_IF_ERROR(dataset_->MakeIterator(ctx.get(), "iterator", &iterator)); in MakeIterator()
114 : dataset_(dataset), in Dataset()
124 Dataset::~Dataset() { dataset_->Unref(); } in ~Dataset()
Dstandalone.h109 DatasetBase* dataset_; // owned variable