Searched refs:finalized_dataset (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
H A D | to_tf_record_op.cc | 88 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 D | random_access_ops.cc | 34 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 D | lookup_ops.cc | 66 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 D | io_ops.cc | 123 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 D | standalone.cc | 109 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 D | standalone.h | 115 Dataset(DatasetBase* finalized_dataset, DatasetBase* original_dataset,
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | reduce_dataset_op.cc | 79 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 D | multi_device_iterator_ops.cc | 669 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 D | iterator_ops.cc | 241 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()
|