/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | indexed_dataset_op.cc | 164 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 …]
|
D | parse_example_dataset_op.cc | 263 : 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/ |
D | parallel_map_dataset_op.cc | 209 : 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 …]
|
D | concatenate_dataset_op_test.cc | 137 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()
|
D | zip_dataset_op_test.cc | 119 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/ |
D | dataset.cc | 41 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/ |
D | standalone.cc | 104 TF_RETURN_IF_ERROR(dataset_->MakeIterator(ctx.get(), "iterator", &iterator)); in MakeIterator() 114 : dataset_(dataset), in Dataset() 124 Dataset::~Dataset() { dataset_->Unref(); } in ~Dataset()
|
D | standalone.h | 109 DatasetBase* dataset_; // owned variable
|