Home
last modified time | relevance | path

Searched refs:finalized_dataset (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dto_tf_record_op.cc88 DatasetBase* finalized_dataset; in DoCompute() local
89 TF_RETURN_IF_ERROR(FinalizeDataset(ctx, dataset, &finalized_dataset)); in DoCompute()
90 core::ScopedUnref unref(finalized_dataset); in DoCompute()
93 TF_RETURN_IF_ERROR(finalized_dataset->MakeIterator( in DoCompute()
96 const int num_output_dtypes = finalized_dataset->output_dtypes().size(); in DoCompute()
102 const DataType dt = finalized_dataset->output_dtypes()[0]; in DoCompute()
H A Drandom_access_ops.cc34 DatasetBase* finalized_dataset; in DoCompute() local
35 TF_ASSIGN_OR_RETURN(finalized_dataset, GetFinalizedDataset(ctx, dataset)); in DoCompute()
42 TF_RETURN_IF_ERROR(finalized_dataset->Get(ctx, index, &components)); in DoCompute()
H A Dlookup_ops.cc66 DatasetBase* finalized_dataset; in Init() local
68 data::FinalizeDataset(ctx, dataset_, &finalized_dataset)); in Init()
69 TF_RETURN_IF_ERROR(finalized_dataset->MakeIterator( in Init()
71 core::ScopedUnref unref(finalized_dataset); in Init()
H A Dio_ops.cc123 DatasetBase* finalized_dataset; in WriteData() local
124 TF_RETURN_IF_ERROR(FinalizeDataset(ctx, dataset, &finalized_dataset)); in WriteData()
127 TF_RETURN_IF_ERROR(finalized_dataset->MakeIterator( in WriteData()
158 finalized_dataset->output_dtypes(), [&mu, &status](Status s) { in WriteData()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dstandalone.cc109 data::DatasetBase* finalized_dataset; in FromGraph() local
117 TF_RETURN_IF_ERROR(data::FinalizeDataset(&ctx, dataset, &finalized_dataset)); in FromGraph()
118 core::ScopedUnref unref(finalized_dataset); in FromGraph()
120 finalized_dataset, dataset, device_mgr.release(), pflr.release(), in FromGraph()
167 Dataset::Dataset(DatasetBase* finalized_dataset, DatasetBase* original_dataset, in Dataset() argument
171 : finalized_dataset_(finalized_dataset), in Dataset()
H A Dstandalone.h115 Dataset(DatasetBase* finalized_dataset, DatasetBase* original_dataset,
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dreduce_dataset_op.cc79 DatasetBase* finalized_dataset = nullptr; in DoCompute() local
80 TF_RETURN_IF_ERROR(FinalizeDataset(ctx, dataset, &finalized_dataset)); in DoCompute()
81 core::ScopedUnref unref(finalized_dataset); in DoCompute()
82 TF_RETURN_IF_ERROR(finalized_dataset->MakeIterator( in DoCompute()
H A Dmulti_device_iterator_ops.cc669 DatasetBase* finalized_dataset; in Compute() local
670 OP_REQUIRES_OK(ctx, FinalizeDataset(ctx, dataset, &finalized_dataset)); in Compute()
671 OP_REQUIRES_OK(ctx, finalized_dataset->MakeIterator(std::move(iter_ctx), in Compute()
674 core::ScopedUnref unref(finalized_dataset); in Compute()
H A Diterator_ops.cc241 DatasetBase* finalized_dataset; in SetIteratorFromDataset() local
242 TF_ASSIGN_OR_RETURN(finalized_dataset, GetFinalizedDataset(ctx, dataset)); in SetIteratorFromDataset()
243 TF_RETURN_IF_ERROR(finalized_dataset->MakeIterator( in SetIteratorFromDataset()