Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/data/
Dtake_dataset_op.cc37 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in TakeDataset()
41 TakeDataset::TakeDataset(DatasetContext::Params params, int64 count, in TakeDataset()
43 : DatasetBase(DatasetContext(std::move(params))), in TakeDataset()
Dtake_dataset_op.h27 TakeDataset(DatasetContext::Params params, int64 count,
Dwindow_dataset.cc33 : DatasetBase(DatasetContext({kWindow})), in WindowDataset()
Dtensor_dataset_op.cc40 : DatasetBase(DatasetContext(ctx)), tensors_(std::move(tensors)) { in Dataset()
Dskip_dataset_op.cc41 : DatasetBase(DatasetContext(ctx)), count_(count), input_(input) { in Dataset()
Dtensor_slice_dataset_op.cc41 : DatasetBase(DatasetContext(ctx)), tensors_(std::move(tensors)) { in Dataset()
Dzip_dataset_op.cc39 : DatasetBase(DatasetContext(ctx)), inputs_(inputs) { in Dataset()
Dconcatenate_dataset_op.cc40 : DatasetBase(DatasetContext(ctx)), in Dataset()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_branch_dataset_op.cc39 WrapperDataset(DatasetContext::Params params, in WrapperDataset()
43 : DatasetBase(DatasetContext(std::move(params))), in WrapperDataset()
210 : DatasetBase(DatasetContext(ctx)), in Dataset()
509 DatasetContext::Params params; in MakeCurrentIterator()
521 DatasetContext::Params take_dataset_params; in MakeCurrentIterator()
Dsnapshot_util.cc360 const int64 start_index, DatasetContext::Params params) in Dataset()
361 : DatasetBase(DatasetContext(std::move(params))), in Dataset()
485 DatasetContext::Params params) in NestedDataset()
486 : DatasetBase(DatasetContext(std::move(params))), datasets_(datasets) { in NestedDataset()
596 DatasetContext::Params({"snapshot_util::Reader::Dataset", in MakeNestedDataset()
610 datasets, DatasetContext::Params({"snapshot_util::Reader::NestedDataset", in MakeNestedDataset()
Dstats_dataset_ops.cc54 : DatasetBase(DatasetContext(ctx)), in Dataset()
177 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dthreadpool_dataset_op.cc144 : DatasetBase(DatasetContext(ctx)), in Dataset()
279 : DatasetBase(DatasetContext(ctx)), in Dataset()
401 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dnon_serializable_dataset_op.cc46 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dignore_errors_dataset_op.cc42 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dsleep_dataset_op.cc46 : DatasetBase(DatasetContext(ctx)), in Dataset()
Drandom_dataset_op.cc41 : DatasetBase(DatasetContext(ctx)), seeds_(seed, seed2) {} in Dataset()
Dassert_cardinality_dataset_op.cc44 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dassert_next_dataset_op.cc40 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dtake_while_dataset_op.cc57 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dunique_dataset_op.cc33 : DatasetBase(DatasetContext(ctx)), input_(input) { in Dataset()
Dsql_dataset_op.cc83 : DatasetBase(DatasetContext(ctx)), in Dataset()
Drebatch_dataset_op.cc60 : DatasetBase(DatasetContext(ctx)), in Dataset()
317 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dset_stats_aggregator_dataset_op.cc107 : DatasetBase(DatasetContext(ctx)), in Dataset()
Dsampling_dataset_op.cc45 : DatasetBase(DatasetContext(ctx)), in Dataset()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h758 class DatasetContext {
766 explicit DatasetContext(Params params) : params_(std::move(params)) {} in DatasetContext() function
768 explicit DatasetContext(OpKernelContext* ctx) { in DatasetContext() function
813 explicit DatasetBase(DatasetContext&& ctx) in DatasetBase()

123