/external/tensorflow/tensorflow/core/kernels/data/ |
D | take_dataset_op.h | 23 class TakeDataset : public DatasetBase { 25 TakeDataset(OpKernelContext* ctx, int64 count, const DatasetBase* input); 28 const DatasetBase* input); 43 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override; 56 const DatasetBase* const input_; 70 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 71 DatasetBase** output) override;
|
D | concatenate_dataset_op.cc | 36 class ConcatenateDatasetOp::Dataset : public DatasetBase { 38 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 39 const DatasetBase* to_concatenate) in Dataset() 40 : DatasetBase(DatasetContext(ctx)), in Dataset() 88 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 212 const DatasetBase* input_; 213 const DatasetBase* to_concatenate_; 220 void ConcatenateDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 221 DatasetBase* to_concatenate, in MakeDataset() 222 DatasetBase** output) { in MakeDataset()
|
D | model_dataset_op.h | 38 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 39 DatasetBase** output) override; 62 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 63 DatasetBase** output) override;
|
D | optimize_dataset_op.h | 49 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 50 DatasetBase** output) override; 73 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 74 DatasetBase** output) override;
|
D | shuffle_dataset_op.h | 47 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 48 DatasetBase** output) override; 66 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 67 DatasetBase** output) override;
|
D | zip_dataset_op.cc | 35 class ZipDatasetOp::Dataset : public DatasetBase { 38 const std::vector<DatasetBase*>& inputs) in Dataset() 39 : DatasetBase(DatasetContext(ctx)), inputs_(inputs) { in Dataset() 90 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 215 const std::vector<DatasetBase*> inputs_; 222 void ZipDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase** output) { in MakeDataset() 223 std::vector<DatasetBase*> inputs; in MakeDataset() 225 DatasetBase* input; in MakeDataset()
|
D | model_dataset_op.cc | 46 class ModelDatasetOp::Dataset : public DatasetBase { 48 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 51 : DatasetBase(DatasetContext(ctx)), in Dataset() 86 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 238 const DatasetBase* input_; 268 void ModelDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 269 DatasetBase** output) { in MakeDataset() 287 void ModelDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 288 DatasetBase** output) { in MakeDataset()
|
D | concatenate_dataset_op.h | 34 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, 35 DatasetBase* to_concatenate, DatasetBase** output) override;
|
D | skip_dataset_op.cc | 38 class SkipDatasetOp::Dataset : public DatasetBase { 40 Dataset(OpKernelContext* ctx, int64 count, const DatasetBase* input) in Dataset() 41 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in Dataset() 78 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 207 const DatasetBase* const input_; 213 void SkipDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 214 DatasetBase** output) { in MakeDataset()
|
D | take_dataset_op.cc | 36 const DatasetBase* input) in TakeDataset() 37 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in TakeDataset() 42 const DatasetBase* input) in TakeDataset() 43 : DatasetBase(DatasetContext(std::move(params))), in TakeDataset() 78 std::vector<const DatasetBase*>* inputs) const { in InputDatasets() 209 void TakeDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 210 DatasetBase** output) { in MakeDataset()
|
D | map_dataset_op.cc | 41 class MapDatasetOp::Dataset : public DatasetBase { 43 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 48 : DatasetBase(DatasetContext(ctx)), in Dataset() 83 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 207 const DatasetBase* const input_; 227 void MapDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 228 DatasetBase** output) { in MakeDataset()
|
D | filter_dataset_op.cc | 48 class FilterDatasetOp::Dataset : public DatasetBase { 50 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 52 : DatasetBase(DatasetContext(ctx)), in Dataset() 78 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 241 const DatasetBase* const input_; 254 void FilterDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 255 DatasetBase** output) { in MakeDataset()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | directed_interleave_dataset_op.cc | 39 class DirectedInterleaveDatasetOp::Dataset : public DatasetBase { 41 Dataset(OpKernelContext* ctx, const DatasetBase* selector_input, in Dataset() 42 std::vector<DatasetBase*> data_inputs) in Dataset() 43 : DatasetBase(DatasetContext(ctx)), in Dataset() 51 const DatasetBase* data_input = data_inputs_[i]; in Dataset() 62 for (DatasetBase* data_input : data_inputs_) { in ~Dataset() 97 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 142 const DatasetBase* data_input = dataset()->data_inputs_[i]; in Initialize() 274 const DatasetBase* const selector_input_; 275 const std::vector<DatasetBase*> data_inputs_; [all …]
|
D | threadpool_dataset_op.cc | 131 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 132 DatasetBase** output) override { in MakeDataset() 140 class Dataset : public DatasetBase { 142 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 144 : DatasetBase(DatasetContext(ctx)), in Dataset() 177 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 251 const DatasetBase* const input_; 262 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 263 DatasetBase** output) override { in MakeDataset() 275 class Dataset : public DatasetBase { [all …]
|
D | stats_dataset_ops.cc | 43 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 44 DatasetBase** output) override { in MakeDataset() 51 class Dataset : public DatasetBase { 53 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag) in Dataset() 54 : DatasetBase(DatasetContext(ctx)), in Dataset() 82 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 156 const DatasetBase* const input_; 166 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 167 DatasetBase** output) override { in MakeDataset() 174 class Dataset : public DatasetBase { [all …]
|
D | non_serializable_dataset_op.cc | 35 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 36 DatasetBase** output) override { in MakeDataset() 41 class Dataset : public DatasetBase { 43 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 46 : DatasetBase(DatasetContext(ctx)), in Dataset() 73 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 131 const DatasetBase* input_;
|
D | ignore_errors_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 33 DatasetBase** output) override { in MakeDataset() 38 class Dataset : public DatasetBase { 40 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 42 : DatasetBase(DatasetContext(ctx)), in Dataset() 70 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 164 const DatasetBase* const input_;
|
D | sleep_dataset_op.cc | 29 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 30 DatasetBase** output) override { in MakeDataset() 42 class Dataset : public DatasetBase { 44 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 46 : DatasetBase(DatasetContext(ctx)), in Dataset() 72 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 167 const DatasetBase* const input_;
|
D | assert_cardinality_dataset_op.cc | 39 class AssertCardinalityDatasetOp::Dataset : public DatasetBase { 41 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 cardinality, in Dataset() 44 : DatasetBase(DatasetContext(ctx)), in Dataset() 71 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 162 const DatasetBase* input_; 176 DatasetBase* input, in MakeDataset() 177 DatasetBase** output) { in MakeDataset()
|
D | assert_next_dataset_op.cc | 34 class AssertNextDatasetOp::Dataset : public DatasetBase { 36 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 40 : DatasetBase(DatasetContext(ctx)), in Dataset() 67 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 146 const DatasetBase* input_; 158 void AssertNextDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 159 DatasetBase** output) { in MakeDataset()
|
D | take_while_dataset_op.cc | 43 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 44 DatasetBase** output) override { in MakeDataset() 53 class Dataset : public DatasetBase { 55 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 57 : DatasetBase(DatasetContext(ctx)), in Dataset() 86 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 204 const DatasetBase* const input_;
|
D | unique_dataset_op.cc | 30 class UniqueDatasetOp::Dataset : public DatasetBase { 32 Dataset(OpKernelContext* ctx, const DatasetBase* input) in Dataset() 33 : DatasetBase(DatasetContext(ctx)), input_(input) { in Dataset() 57 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 210 const DatasetBase* const input_; 213 void UniqueDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 214 DatasetBase** output) { in MakeDataset()
|
D | set_stats_aggregator_dataset_op.cc | 86 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 87 DatasetBase** output) override { in MakeDataset() 101 class Dataset : public DatasetBase { 103 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset() 107 : DatasetBase(DatasetContext(ctx)), in Dataset() 142 std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 225 const DatasetBase* const input_;
|
D | sampling_dataset_op.cc | 41 class SamplingDatasetOp::Dataset : public DatasetBase { 44 const DatasetBase* input) in Dataset() 45 : DatasetBase(DatasetContext(ctx)), in Dataset() 73 Status InputDatasets(std::vector<const DatasetBase*>* inputs) const override { in InputDatasets() 219 const DatasetBase* const input_; 226 void SamplingDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset() 227 DatasetBase** output) { in MakeDataset()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.cc | 59 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {} in DatasetVariantWrapper() 78 DatasetBase* get() const { return dataset_; } in get() 99 DatasetBase* dataset_; // Owns one reference. 144 DatasetBase* unused; in Compute() 212 const DatasetBase* dataset, in AddDataset() 360 DatasetBase* dataset; in GetAllocatedBytes() 374 DatasetBase* dataset; in GetTotalBytes() 395 DatasetBase** out_dataset) { in GetDatasetFromVariantTensor() 413 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) { in StoreDatasetInVariantTensor() 423 Status DatasetBase::MakeIterator( in MakeIterator() [all …]
|