| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 154 class Iterator : public DatasetIterator<Dataset<T>> {
157 : DatasetIterator<Dataset<T>>(params) {} in Iterator()
160 return DatasetIterator<Dataset<T>>::dataset()->input_->MakeIterator( in Initialize()
161 ctx, this, DatasetIterator<Dataset<T>>::prefix(), &input_impl_); in Initialize()
172 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in GetNextInternal()
174 DatasetIterator<Dataset<T>>::dataset()->row_shape_; in GetNextInternal()
193 i < DatasetIterator<Dataset<T>>::dataset()->batch_size_ && in GetNextInternal()
243 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0], in GetNextInternal()
295 DatasetIterator<Dataset<T>>::dataset()->batch_size_); in CreateNode()
|
| D | lookup_ops.cc | 49 class DatasetIterator class
52 explicit DatasetIterator(data::DatasetBase* dataset) : dataset_(dataset) {} in DatasetIterator() function in tensorflow::data::experimental::__anonb6ed0e5a0111::DatasetIterator
54 ~DatasetIterator() override {} in ~DatasetIterator()
171 DatasetIterator iter(dataset); in InitializeTableFromDataset()
|
| D | stats_dataset_ops.cc | 106 class Iterator : public DatasetIterator<Dataset> {
109 : DatasetIterator<Dataset>(params) {} in Iterator()
225 class Iterator : public DatasetIterator<Dataset> {
228 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | threadpool_dataset_op.cc | 227 class Iterator : public DatasetIterator<Dataset> {
230 : DatasetIterator<Dataset>(params) {} in Iterator()
348 class Iterator : public DatasetIterator<Dataset> {
351 : DatasetIterator<Dataset>(params) {} in Iterator()
488 class Iterator : public DatasetIterator<Dataset> {
491 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | non_serializable_dataset_op.cc | 95 class Iterator : public DatasetIterator<Dataset> {
98 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | ignore_errors_dataset_op.cc | 94 class Iterator : public DatasetIterator<Dataset> {
97 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | sleep_dataset_op.cc | 103 class Iterator : public DatasetIterator<Dataset> {
106 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | snapshot_dataset_op.cc | 254 class Reader : public DatasetIterator<Dataset> {
259 : DatasetIterator<Dataset>(params), start_index_(start_index) {} in Reader()
346 class Writer : public DatasetIterator<Dataset> {
354 : DatasetIterator<Dataset>(params), in Writer()
563 class Passthrough : public DatasetIterator<Dataset> {
568 : DatasetIterator<Dataset>(params) {} in Passthrough()
595 class Iterator : public DatasetIterator<Dataset> {
603 : DatasetIterator<Dataset>(params), in Iterator()
1106 class Iterator : public DatasetIterator<Dataset> {
1109 : DatasetIterator<Dataset>(params) { in Iterator()
[all …]
|
| D | assert_next_dataset_op.cc | 90 class Iterator : public DatasetIterator<Dataset> {
93 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | assert_cardinality_dataset_op.cc | 94 class Iterator : public DatasetIterator<Dataset> {
97 : DatasetIterator<Dataset>(params), num_elements_(0) {} in Iterator()
|
| D | random_dataset_op.cc | 101 class Iterator : public DatasetIterator<Dataset> {
104 : DatasetIterator<Dataset>(params), in Iterator()
|
| D | take_while_dataset_op.cc | 123 class Iterator : public DatasetIterator<Dataset> {
126 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | choose_fastest_branch_dataset_op.cc | 84 class WrapperIterator : public DatasetIterator<WrapperDataset> {
87 : DatasetIterator<WrapperDataset>(params), error_(error) {} in WrapperIterator()
337 class ChooseFastestIterator : public DatasetIterator<Dataset> {
340 : DatasetIterator<Dataset>(params), in ChooseFastestIterator()
|
| D | rebatch_dataset_op.cc | 118 class Iterator : public DatasetIterator<Dataset> {
121 : DatasetIterator<Dataset>(params) {} in Iterator()
381 class Iterator : public DatasetIterator<Dataset> {
384 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | unique_dataset_op.cc | 77 class Iterator : public DatasetIterator<Dataset> {
80 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| D | sampling_dataset_op.cc | 100 class Iterator : public DatasetIterator<Dataset> {
103 : DatasetIterator<Dataset>(params), in Iterator()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | repeat_dataset_op.cc | 139 class EmptyIterator : public DatasetIterator<Dataset> {
142 : DatasetIterator<Dataset>(params) {} in EmptyIterator()
167 class FiniteIterator : public DatasetIterator<Dataset> {
170 : DatasetIterator<Dataset>(params), i_(0) {} in FiniteIterator()
239 class ForeverIterator : public DatasetIterator<Dataset> {
242 : DatasetIterator<Dataset>(params), in ForeverIterator()
|
| D | take_dataset_op.cc | 107 class TakeDataset::EmptyIterator : public DatasetIterator<TakeDataset> {
110 : DatasetIterator<TakeDataset>(params) {} in EmptyIterator()
135 class TakeDataset::FiniteIterator : public DatasetIterator<TakeDataset> {
138 : DatasetIterator<TakeDataset>(params), i_(0) {} in FiniteIterator()
|
| D | skip_dataset_op.cc | 114 class EmptyIterator : public DatasetIterator<Dataset> {
117 : DatasetIterator<Dataset>(params) {} in EmptyIterator()
143 class FiniteIterator : public DatasetIterator<Dataset> {
146 : DatasetIterator<Dataset>(params), i_(0) {} in FiniteIterator()
|
| D | cache_dataset_ops.cc | 201 class FileIterator : public DatasetIterator<FileDatasetBase> {
204 : DatasetIterator<FileDatasetBase>(params) { in FileIterator()
282 class FileWriterIterator : public DatasetIterator<FileDatasetBase> {
285 : DatasetIterator<FileDatasetBase>(params), in FileWriterIterator()
552 class FileReaderIterator : public DatasetIterator<FileDatasetBase> {
555 : DatasetIterator<FileDatasetBase>(params), in FileReaderIterator()
797 class MemoryIterator : public DatasetIterator<MemoryDatasetBase> {
800 : DatasetIterator<MemoryDatasetBase>(params), cache_(cache) {} in MemoryIterator()
848 class MemoryWriterIterator : public DatasetIterator<MemoryDatasetBase> {
851 : DatasetIterator<MemoryDatasetBase>(params), cache_(cache) {} in MemoryWriterIterator()
[all …]
|
| D | window_dataset.cc | 108 class Iterator : public DatasetIterator<Window> {
110 explicit Iterator(const Params& params) : DatasetIterator<Window>(params) {} in Iterator()
|
| D | tensor_dataset_op.cc | 119 class Iterator : public DatasetIterator<Dataset> {
122 : DatasetIterator<Dataset>(params), produced_(false) {} in Iterator()
|
| D | fixed_length_record_dataset_op.cc | 128 class UncompressedIterator : public DatasetIterator<Dataset> {
131 : DatasetIterator<Dataset>(params) {} in UncompressedIterator()
254 class CompressedIterator : public DatasetIterator<Dataset> {
257 : DatasetIterator<Dataset>(params) {} in CompressedIterator()
|
| D | generator_dataset_op.cc | 97 class Iterator : public DatasetIterator<Dataset> {
100 : DatasetIterator<Dataset>(params) {} in Iterator()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_side_effects.h | 56 struct DatasetIterator : ::mlir::SideEffects::Resource::Base<DatasetIterator> { struct
|