Home
last modified time | relevance | path

Searched defs:DatasetV2 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.h52 class DatasetV2; variable
71 class DatasetV2; variable
Dshuffle_dataset_op.cc577 class ShuffleDatasetOp::DatasetV2 : public ShuffleDatasetBase { class in tensorflow::data::ShuffleDatasetOp
579 DatasetV2(OpKernelContext* ctx, const DatasetBase* input, int64_t buffer_size, in DatasetV2() function in tensorflow::data::ShuffleDatasetOp::DatasetV2
880 class ShuffleAndRepeatDatasetOp::DatasetV2 : public ShuffleDatasetBase { class in tensorflow::data::ShuffleAndRepeatDatasetOp
882 DatasetV2(OpKernelContext* ctx, const DatasetBase* input, int64_t buffer_size, in DatasetV2() function in tensorflow::data::ShuffleAndRepeatDatasetOp::DatasetV2
/external/tensorflow/tensorflow/dtensor/python/
Dinput_util.py106 def __init__(self, datasets: Sequence[Tuple[int, dataset_ops.DatasetV2]],
338 dataset: dataset_ops.DatasetV2,
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py136 class DatasetV2( class