Home
last modified time | relevance | path

Searched refs:DatasetContext (Results 1 – 25 of 67) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/data/
Dtake_dataset_op.h28 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in TakeDataset()
32 TakeDataset(DatasetContext::Params params, int64 count, in TakeDataset()
34 : DatasetBase(DatasetContext(std::move(params))), in TakeDataset()
Dgraph_rewrite_dataset.h39 : DatasetBase(DatasetContext(ctx)), in GraphRewriteDataset()
Dwindow_dataset.cc27 : DatasetBase(DatasetContext({"Window"})), in WindowDataset()
Dfilter_by_component_dataset_op.cc56 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dtensor_dataset_op.cc50 : DatasetBase(DatasetContext(ctx)), tensors_(std::move(tensors)) { in Dataset()
Drange_dataset_op.cc49 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dshard_dataset_op.cc59 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dzip_dataset_op.cc45 : DatasetBase(DatasetContext(ctx)), inputs_(inputs) { in Dataset()
Dtensor_slice_dataset_op.cc67 : DatasetBase(DatasetContext(ctx)), tensors_(std::move(tensors)) { in Dataset()
Dgenerator_dataset_op.cc38 : DatasetBase(DatasetContext(ctx)), in Dataset()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_branch_dataset_op.cc37 WrapperDataset(DatasetContext::Params params, in WrapperDataset()
41 : DatasetBase(DatasetContext(std::move(params))), in WrapperDataset()
196 : DatasetBase(DatasetContext(ctx)), in Dataset()
482 DatasetContext::Params params; in MakeCurrentIterator()
494 DatasetContext::Params take_dataset_params; in MakeCurrentIterator()
Dthreadpool_dataset_op.cc143 : DatasetBase(DatasetContext(ctx)), in Dataset()
255 : DatasetBase(DatasetContext(ctx)), in Dataset()
352 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dstats_dataset_ops.cc53 : DatasetBase(DatasetContext(ctx)), in Dataset()
163 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dnon_serializable_dataset_op.cc45 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dsleep_dataset_op.cc47 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dignore_errors_dataset_op.cc40 : DatasetBase(DatasetContext(ctx)), input_(input) { in Dataset()
Dassert_next_dataset_op.cc52 : DatasetBase(DatasetContext(ctx)), in Dataset()
Drandom_dataset_op.cc55 : DatasetBase(DatasetContext(ctx)), seed_(seed), seed2_(seed2) {} in Dataset()
Dindexed_dataset_op.cc64 explicit IndexedDataset(DatasetContext&& ctx) : DatasetBase(std::move(ctx)) {} in IndexedDataset()
422 : IndexedDataset(DatasetContext(ctx)), size_(size) {} in Dataset()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h545 class DatasetContext {
553 explicit DatasetContext(Params params) : params_(std::move(params)) {} in DatasetContext() function
555 explicit DatasetContext(OpKernelContext* ctx) { in DatasetContext() function
597 explicit DatasetBase(DatasetContext&& ctx) in DatasetBase()
947 using data::DatasetContext;
/external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/
Dignite_dataset.cc29 : DatasetBase(DatasetContext(ctx)), in IgniteDataset()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_sample_keys_dataset_op.cc39 : DatasetBase(DatasetContext(ctx)), table_(table) { in Dataset()
Dbigtable_prefix_key_dataset_op.cc44 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dbigtable_range_key_dataset_op.cc48 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dbigtable_sample_key_pairs_dataset_op.cc61 : DatasetBase(DatasetContext(ctx)), in Dataset()

123