| /external/tensorflow/tensorflow/core/kernels/data/ |
| window_dataset.h | 44 DatasetBase** out_dataset);
|
| dataset_utils.cc | 38 DatasetBase* returned_dataset;
|
| stats_dataset_ops.cc | 42 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
43 DatasetBase** output) override {
52 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag)
126 const DatasetBase* const input_;
136 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
137 DatasetBase** output) override {
146 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input, string tag)
224 const DatasetBase* const input_;
|
| zip_dataset_op.cc | 30 void MakeDataset(OpKernelContext* ctx, DatasetBase** output) override {
31 std::vector<DatasetBase*> inputs;
33 DatasetBase* input;
44 const std::vector<DatasetBase*>& inputs)
165 const std::vector<DatasetBase*> inputs_;
|
| concatenate_dataset_op.cc | 30 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
31 DatasetBase* to_concatenate, DatasetBase** output) override {
44 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input,
45 const DatasetBase* to_concatenate)
183 const DatasetBase* input_;
184 const DatasetBase* to_concatenate_;
|
| skip_dataset_op.cc | 31 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
32 DatasetBase** output) override {
43 Dataset(OpKernelContext* ctx, int64 count, const DatasetBase* input)
183 const DatasetBase* const input_;
|
| take_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
33 DatasetBase** output) override {
43 Dataset(OpKernelContext* ctx, int64 count, const DatasetBase* input)
170 const DatasetBase* const input_;
|
| cache_dataset_ops.cc | 34 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
35 DatasetBase** output) override {
49 class FileDataset : public DatasetBase {
51 explicit FileDataset(const DatasetBase* input, string filename, Env* env)
253 const DatasetBase* const input_;
263 class MemoryDataset : public DatasetBase {
265 explicit MemoryDataset(const DatasetBase* input) : input_(input) {
382 const DatasetBase* const input_;
|
| window_dataset.cc | 20 class WindowDataset : public DatasetBase {
91 DatasetBase** out_dataset) {
|
| batch_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
33 DatasetBase** output) override {
47 Dataset(OpKernelContext* ctx, int64 batch_size, const DatasetBase* input)
201 const DatasetBase* const input_;
|
| filter_dataset_op.cc | 37 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
38 DatasetBase** output) override {
59 Dataset(OpKernelContext* ctx, const DatasetBase* input,
193 const DatasetBase* const input_;
|
| map_dataset_op.cc | 39 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
40 DatasetBase** output) override {
60 Dataset(OpKernelContext* ctx, const DatasetBase* input,
173 const DatasetBase* const input_;
|
| repeat_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
33 DatasetBase** output) override {
44 Dataset(OpKernelContext* ctx, int64 count, const DatasetBase* input)
229 const DatasetBase* const input_;
|
| unique_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
33 DatasetBase** output) override {
52 Dataset(OpKernelContext* ctx, const DatasetBase* input)
210 const DatasetBase* const input_;
|
| shuffle_dataset_op.cc | 44 ShuffleDatasetBase(OpKernelContext* ctx, const DatasetBase* input,
300 const DatasetBase* const input_;
314 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
315 DatasetBase** output) override {
351 ReshufflingDataset(OpKernelContext* ctx, const DatasetBase* input,
391 FixedSeedDataset(OpKernelContext* ctx, const DatasetBase* input,
443 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
444 DatasetBase** output) override {
474 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 buffer_size,
|
| sql_dataset_ops.cc | 53 void MakeDataset(OpKernelContext* ctx, DatasetBase** output) override {
78 class Dataset : public DatasetBase {
|
| dense_to_sparse_batch_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
33 DatasetBase** output) override {
82 const PartialTensorShape& row_shape, const DatasetBase* input)
290 const DatasetBase* const input_;
|
| flat_map_dataset_op.cc | 40 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
41 DatasetBase** output) override {
61 Dataset(OpKernelContext* ctx, const DatasetBase* input,
264 const DatasetBase* const input_;
|
| prefetch_dataset_op.cc | 35 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
36 DatasetBase** output) override {
49 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 buffer_size)
332 const DatasetBase* const input_;
|
| scan_dataset_op.cc | 43 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
44 DatasetBase** output) override {
74 Dataset(OpKernelContext* ctx, const DatasetBase* input,
257 const DatasetBase* const input_;
|
| map_and_batch_dataset_op.cc | 46 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
47 DatasetBase** output) override {
79 class Dataset : public DatasetBase {
81 Dataset(const DatasetBase* input, int64 batch_size,
387 const DatasetBase* const input_;
|
| parallel_interleave_dataset_op.cc | 43 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
44 DatasetBase** output) override {
93 class Dataset : public DatasetBase {
95 Dataset(const DatasetBase* input,
481 const DatasetBase* const input_;
|
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| ignore_errors_dataset_op.cc | 32 void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
33 DatasetBase** output) override {
40 explicit Dataset(OpKernelContext* ctx, const DatasetBase* input)
126 const DatasetBase* const input_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| dataset.cc | 24 // A wrapper class for storing a `DatasetBase` instance in a DT_VARIANT tensor.
25 // Objects of the wrapper class own a reference on an instance of `DatasetBase`,
31 // `DatasetBase` object, so the `Encode()` and `Decode()` methods are not
38 explicit DatasetVariantWrapper(DatasetBase* dataset) : dataset_(dataset) {}
49 DatasetBase* get() const { return dataset_; }
70 DatasetBase* const dataset_; // Owns one reference.
213 DatasetBase** out_dataset) {
231 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) {
242 DatasetBase* dataset = nullptr;
252 DatasetBase** output)
[all...] |
| dataset.h | 313 // range of outputs is typically represented by an `DatasetBase`,
387 class DatasetBase : public core::RefCounted {
419 return errors::Unimplemented("DatasetBase::Save");
430 Status AddParentDataset(OpKernelContext* ctx, const DatasetBase* dataset,
449 class GraphDatasetBase : public DatasetBase {
548 // Encapsulates the work required to plug a DatasetBase into the core TensorFlow
558 virtual void MakeDataset(OpKernelContext* ctx, DatasetBase** output) = 0;
580 void MakeDataset(OpKernelContext* ctx, DatasetBase** output) final;
581 virtual void MakeDataset(OpKernelContext* ctx, DatasetBase* input,
582 DatasetBase** output) = 0
[all...] |
