Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/framework/
Ddataset.cc44 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {} in DatasetVariantWrapper()
55 DatasetBase* get() const { return dataset_; } in get()
76 DatasetBase* const dataset_; // Owns one reference.
121 DatasetBase* unused; in Compute()
189 const DatasetBase* dataset, in AddDataset()
321 DatasetBase* dataset; in GetAllocatedBytes()
334 DatasetBase** out_dataset) { in GetDatasetFromVariantTensor()
352 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) { in StoreDatasetInVariantTensor()
362 Status DatasetBase::Save(SerializationContext* ctx, in Save()
381 Status DatasetBase::DatasetGraphDefBuilder::AddInputDataset( in AddInputDataset()
[all …]
Ddataset.h59 class DatasetBase; variable
151 Status AddDataset(const DatasetBase* dataset, in AddDataset()
164 Status AddDataset(const DatasetBase* dataset, in AddDataset()
176 const DatasetBase* dataset,
527 friend class DatasetBase; // for access to `AddCleanupFunction`
579 DatasetBase** out_dataset);
584 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor);
588 class DatasetBase : public core::RefCounted {
597 explicit DatasetBase(DatasetContext&& ctx) in DatasetBase() function
665 const DatasetBase* dataset, Node** output);
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc128 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
129 DatasetBase** output) override { in MakeDataset()
139 class Dataset : public DatasetBase {
141 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
143 : DatasetBase(DatasetContext(ctx)), in Dataset()
227 const DatasetBase* const input_;
238 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
239 DatasetBase** output) override { in MakeDataset()
251 class Dataset : public DatasetBase {
253 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
[all …]
Ddirected_interleave_dataset_op.cc32 void MakeDataset(OpKernelContext* ctx, DatasetBase** output) override { in MakeDataset()
33 DatasetBase* selector_input; in MakeDataset()
47 std::vector<DatasetBase*> data_inputs; in MakeDataset()
49 DatasetBase* input; in MakeDataset()
66 class Dataset : public DatasetBase {
68 Dataset(OpKernelContext* ctx, const DatasetBase* selector_input, in Dataset()
69 std::vector<DatasetBase*> data_inputs) in Dataset()
70 : DatasetBase(DatasetContext(ctx)), in Dataset()
78 const DatasetBase* data_input = data_inputs_[i]; in Dataset()
89 for (DatasetBase* data_input : data_inputs_) { in ~Dataset()
[all …]
Dstats_dataset_ops.cc42 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
43 DatasetBase** output) override { in MakeDataset()
50 class Dataset : public DatasetBase {
52 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag) in Dataset()
53 : DatasetBase(DatasetContext(ctx)), in Dataset()
142 const DatasetBase* const input_;
152 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
153 DatasetBase** output) override { in MakeDataset()
160 class Dataset : public DatasetBase {
162 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag) in Dataset()
[all …]
Dnon_serializable_dataset_op.cc34 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
35 DatasetBase** output) override { in MakeDataset()
40 class Dataset : public DatasetBase {
42 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
45 : DatasetBase(DatasetContext(ctx)), in Dataset()
117 const DatasetBase* input_;
Dsleep_dataset_op.cc30 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
31 DatasetBase** output) override { in MakeDataset()
43 class Dataset : public DatasetBase {
45 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
47 : DatasetBase(DatasetContext(ctx)), in Dataset()
130 const DatasetBase* const input_;
Dignore_errors_dataset_op.cc31 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
32 DatasetBase** output) override { in MakeDataset()
37 class Dataset : public DatasetBase {
39 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input) in Dataset()
40 : DatasetBase(DatasetContext(ctx)), input_(input) { in Dataset()
139 const DatasetBase* const input_;
Dassert_next_dataset_op.cc36 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
37 DatasetBase** output) override { in MakeDataset()
46 class Dataset : public DatasetBase {
48 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
52 : DatasetBase(DatasetContext(ctx)), in Dataset()
148 const DatasetBase* input_;
Dtake_while_dataset_op.cc49 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
50 DatasetBase** output) override { in MakeDataset()
92 class Dataset : public DatasetBase {
94 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
98 : DatasetBase(DatasetContext(ctx)), in Dataset()
143 DatasetBase* input; in AsGraphDefInternal()
236 const DatasetBase* const input_;
Dset_stats_aggregator_dataset_op.cc79 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
80 DatasetBase** output) override { in MakeDataset()
95 class Dataset : public DatasetBase {
97 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
101 : DatasetBase(DatasetContext(ctx)), in Dataset()
201 const DatasetBase* const input_;
Dunique_dataset_op.cc32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
33 DatasetBase** output) override { in MakeDataset()
50 class Dataset : public DatasetBase {
52 Dataset(OpKernelContext* ctx, const DatasetBase* input) in Dataset()
53 : DatasetBase(DatasetContext(ctx)), input_(input) { in Dataset()
220 const DatasetBase* const input_;
Dunbatch_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, DatasetBase* input) in Dataset()
41 : DatasetBase(DatasetContext(ctx)), input_(input) { in Dataset()
219 const DatasetBase* const input_;
/external/tensorflow/tensorflow/core/kernels/data/
Dtake_dataset_op.h25 class TakeDataset : public DatasetBase {
27 TakeDataset(OpKernelContext* ctx, int64 count, const DatasetBase* input) in TakeDataset()
28 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in TakeDataset()
33 const DatasetBase* input) in TakeDataset()
34 : DatasetBase(DatasetContext(std::move(params))), in TakeDataset()
75 const DatasetBase* const input_;
Dconcatenate_dataset_op.cc30 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
31 DatasetBase* to_concatenate, DatasetBase** output) override { in MakeDataset()
42 class Dataset : public DatasetBase {
44 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
45 const DatasetBase* to_concatenate) in Dataset()
46 : DatasetBase(DatasetContext(ctx)), in Dataset()
206 const DatasetBase* input_;
207 const DatasetBase* to_concatenate_;
Dmap_dataset_op_test.cc67 DatasetBase* const input_dataset, OpKernel* const map_kernel, in CreateMapDatasetContext()
119 DatasetBase* range_dataset; in TEST_P()
131 DatasetBase* map_dataset; in TEST_P()
177 DatasetBase* range_dataset; in TEST_F()
188 DatasetBase* map_dataset; in TEST_F()
204 DatasetBase* range_dataset; in TEST_F()
215 DatasetBase* map_dataset; in TEST_F()
232 DatasetBase* range_dataset; in TEST_F()
243 DatasetBase* map_dataset; in TEST_F()
283 DatasetBase* range_dataset; in TEST_P()
[all …]
Dgraph_rewrite_dataset.h34 class GraphRewriteDataset : public DatasetBase {
36 GraphRewriteDataset(OpKernelContext* ctx, const DatasetBase* input, in GraphRewriteDataset()
39 : DatasetBase(DatasetContext(ctx)), in GraphRewriteDataset()
82 DatasetBase* optimized_input_;
87 const DatasetBase* input_;
Dfilter_by_component_dataset_op.cc41 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
42 DatasetBase** output) override { in MakeDataset()
51 class Dataset : public DatasetBase {
53 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
56 : DatasetBase(DatasetContext(ctx)), in Dataset()
96 const DatasetBase* const input_;
Dzip_dataset_op.cc30 void MakeDataset(OpKernelContext* ctx, DatasetBase** output) override { in MakeDataset()
31 std::vector<DatasetBase*> inputs; in MakeDataset()
33 DatasetBase* input; in MakeDataset()
41 class Dataset : public DatasetBase {
44 const std::vector<DatasetBase*>& inputs) in Dataset()
45 : DatasetBase(DatasetContext(ctx)), inputs_(inputs) { in Dataset()
192 const std::vector<DatasetBase*> inputs_;
Dshard_dataset_op.cc32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
33 DatasetBase** output) override { in MakeDataset()
55 class Dataset : public DatasetBase {
58 const DatasetBase* input) in Dataset()
59 : DatasetBase(DatasetContext(ctx)), in Dataset()
186 const DatasetBase* const input_;
Dmodel_dataset_op.cc44 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
45 DatasetBase** output) override { in MakeDataset()
50 class Dataset : public DatasetBase {
52 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 cpu_budget) in Dataset()
53 : DatasetBase(DatasetContext(ctx)), in Dataset()
198 const DatasetBase* input_;
Dskip_dataset_op.cc31 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
32 DatasetBase** output) override { in MakeDataset()
41 class Dataset : public DatasetBase {
43 Dataset(OpKernelContext* ctx, int64 count, const DatasetBase* input) in Dataset()
44 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in Dataset()
203 const DatasetBase* const input_;
Dflat_map_dataset_op.cc40 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
41 DatasetBase** output) override { in MakeDataset()
50 class Dataset : public DatasetBase {
52 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
57 : DatasetBase(DatasetContext(ctx)), in Dataset()
98 DatasetBase* input; in AsGraphDefInternal()
264 const DatasetBase* const input_;
Dfilter_dataset_op.cc50 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
51 DatasetBase** output) override { in MakeDataset()
94 class Dataset : public DatasetBase {
96 Dataset(OpKernelContext* ctx, const DatasetBase* input, in Dataset()
100 : DatasetBase(DatasetContext(ctx)), in Dataset()
140 DatasetBase* input; in AsGraphDefInternal()
288 const DatasetBase* const input_;
Dbatch_dataset_op.cc33 void MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
34 DatasetBase** output) override { in MakeDataset()
52 class Dataset : public DatasetBase {
55 const DatasetBase* input) in Dataset()
56 : DatasetBase(DatasetContext(ctx)), in Dataset()
250 const DatasetBase* const input_;

1234