Home
last modified time | relevance | path

Searched refs:InputDatasets (Results 1 – 25 of 73) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dfinalize_dataset_op_test.cc70 TF_ASSERT_OK(dataset->InputDatasets(&input_datasets)); in CheckDatasetPipelineTypeString()
244 Status s = dataset_->InputDatasets(&inputs); in ITERATOR_GET_NEXT_TEST_P()
254 Status s = dataset_->InputDatasets(&inputs); in TEST_F()
264 Status s = dataset_->InputDatasets(&inputs); in TEST_F()
274 Status s = dataset_->InputDatasets(&inputs); in TEST_F()
285 Status s = dataset_->InputDatasets(&inputs); in TEST_F()
H A Dtake_dataset_op.h45 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override;
H A Doptions_dataset_op.cc85 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::OptionsDatasetOp::Dataset
H A Dwindow_dataset.cc77 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::__anon2a1301800111::Window
H A Dtensor_dataset_op.cc81 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::TensorDatasetOp::Dataset
H A Dgenerator_dataset_op.cc78 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::GeneratorDatasetOp::Dataset
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dsplit_utils.cc135 TF_RETURN_IF_ERROR(dataset->InputDatasets(&inputs)); in GetSplitProviders()
149 TF_RETURN_IF_ERROR(dataset->InputDatasets(&inputs)); in CreateInputIteratorContexts()
200 TF_RETURN_IF_ERROR(dataset->InputDatasets(&inputs)); in GetSplitProviders()
214 TF_RETURN_IF_ERROR(dataset->InputDatasets(&inputs)); in CreateInputIteratorContexts()
H A Droot_dataset.cc339 TF_RETURN_IF_ERROR(this->InputDatasets(&inputs)); in Get()
343 Status RootDataset::InputDatasets( in InputDatasets() function in tensorflow::data::RootDataset
H A Droot_dataset.h54 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override;
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset.cc596 Status s = InputDatasets(&inputs); in ComputeNumSources()
660 Status s = InputDatasets(&inputs); in MergeOptionsFromInputs()
688 Status s = InputDatasets(&inputs); in MakeIterator()
712 Status s = InputDatasets(&inputs); in MakeSplitProviders()
745 Status DatasetBase::InputDatasets( in InputDatasets() function in tensorflow::data::DatasetBase
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dthreadpool_dataset_op.cc203 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::ThreadPoolDatasetOp::Dataset
323 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::MaxIntraOpParallelismDatasetOp::Dataset
464 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::PrivateThreadPoolDatasetOp::Dataset
H A Dassert_prev_dataset_op.cc63 TF_RETURN_IF_ERROR(dataset.InputDatasets(&inputs)); in GetPreviousDataset()
166 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::AssertPrevDatasetOp::Dataset
H A Dchoose_fastest_branch_dataset_op.cc58 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::__anon50b30f2a0111::WrapperDataset
252 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anon50b30f2a0111::ChooseFastestBranchDatasetOp::Dataset
H A Drebatch_dataset_op.cc94 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anondafc533c0111::RebatchDatasetOp::Dataset
355 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anondafc533c0111::RebatchDatasetV2Op::Dataset
H A Dnon_serializable_dataset_op.cc72 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anon6c78f5640111::NonSerializableDatasetOp::Dataset
H A Dignore_errors_dataset_op.cc69 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anonbd5747c30111::IgnoreErrorsDatasetOp::Dataset
H A Dsleep_dataset_op.cc73 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anon5a0a93fc0111::SleepDatasetOp::Dataset
H A Drandom_dataset_op.cc82 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::RandomDatasetOp::Dataset
H A Dassert_next_dataset_op.cc67 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::AssertNextDatasetOp::Dataset
H A Dassert_cardinality_dataset_op.cc71 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::AssertCardinalityDatasetOp::Dataset
H A Dtake_while_dataset_op.cc85 Status InputDatasets( in InputDatasets() function in tensorflow::data::experimental::__anon28213f200111::TakeWhileDatasetOp::Dataset
H A Dunique_dataset_op.cc57 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::UniqueDatasetOp::Dataset
H A Dsampling_dataset_op.cc73 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::experimental::SamplingDatasetOp::Dataset
H A Dlist_dataset_op.cc86 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() function in tensorflow::data::ListDatasetOp::Dataset
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dexternal_dataset_op.cc125 tensorflow::Status InputDatasets( in InputDatasets() function in fcp::ExternalDatasetOp::Dataset

123