Home
last modified time | relevance | path

Searched refs:DatasetBase (Results 1 – 25 of 131) sorted by relevance

123456

/external/tensorflow/tensorflow/core/kernels/data/
Dtake_dataset_op.h23 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;
Dconcatenate_dataset_op.cc36 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()
Dmodel_dataset_op.h38 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
39 DatasetBase** output) override;
62 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
63 DatasetBase** output) override;
Doptimize_dataset_op.h49 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
50 DatasetBase** output) override;
73 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
74 DatasetBase** output) override;
Dshuffle_dataset_op.h47 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
48 DatasetBase** output) override;
66 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
67 DatasetBase** output) override;
Dzip_dataset_op.cc35 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()
Dmodel_dataset_op.cc46 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()
Dconcatenate_dataset_op.h34 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
35 DatasetBase* to_concatenate, DatasetBase** output) override;
Dskip_dataset_op.cc38 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()
Dtake_dataset_op.cc36 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()
Dmap_dataset_op.cc41 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()
Dfilter_dataset_op.cc48 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/
Ddirected_interleave_dataset_op.cc39 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 …]
Dthreadpool_dataset_op.cc131 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 …]
Dstats_dataset_ops.cc43 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 …]
Dnon_serializable_dataset_op.cc35 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_;
Dignore_errors_dataset_op.cc32 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_;
Dsleep_dataset_op.cc29 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_;
Dassert_cardinality_dataset_op.cc39 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()
Dassert_next_dataset_op.cc34 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()
Dtake_while_dataset_op.cc43 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_;
Dunique_dataset_op.cc30 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()
Dset_stats_aggregator_dataset_op.cc86 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_;
Dsampling_dataset_op.cc41 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/
Ddataset.cc59 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 …]

123456