Searched defs:DatasetV2 (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | shuffle_dataset_op.h | 52 class DatasetV2; variable
71 class DatasetV2; variable
|
| H A D | shuffle_dataset_op.cc | 577 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/ |
| H A D | input_util.py | 106 def __init__(self, datasets: Sequence[Tuple[int, dataset_ops.DatasetV2]],
338 dataset: dataset_ops.DatasetV2,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 136 class DatasetV2( class
|