Home
last modified time | relevance | path

Searched defs:input_dataset (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py4388 def __init__(self, input_dataset, variant_tensor): argument
4399 def __init__(self, input_dataset, variant_tensor): argument
4878 def __init__(self, input_dataset, dataset_to_concatenate, name=None): argument
4919 def __init__(self, input_dataset, count, name=None): argument
4983 def __init__(self, input_dataset, filename, name=None): argument
5007 input_dataset, argument
5046 def __init__(self, input_dataset, count, name=None): argument
5061 def __init__(self, input_dataset, count, name=None): argument
5076 def __init__(self, input_dataset, num_shards, index, name=None): argument
5094 def __init__(self, input_dataset, batch_size, drop_remainder, name=None): argument
[all …]
H A Dreaders.py321 input_dataset, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dtesting.py121 def __init__(self, input_dataset, transformations): argument
141 def __init__(self, input_dataset, transformations): argument
178 def __init__(self, input_dataset): argument
191 def __init__(self, input_dataset, sleep_microseconds): argument
H A Ddistribute.py70 def __init__(self, input_dataset, num_workers, index, num_replicas=None): argument
89 def _AutoShardDatasetV1(input_dataset, num_workers, index, num_replicas=None): # pylint: disable=i… argument
121 def __init__(self, input_dataset, batch_sizes, drop_remainder=False): argument
258 def __init__(self, input_dataset, num_replicas): argument
H A Dgrouping.py263 def __init__(self, input_dataset, key_func, reducer): argument
283 def _make_key_func(self, key_func, input_dataset): argument
303 def _make_reduce_func(self, reduce_func, input_dataset): argument
H A Dbatching.py304 def __init__(self, input_dataset, batch_size, row_shape): argument
335 def __init__(self, input_dataset, map_func, batch_size, num_parallel_calls, argument
396 def __init__(self, input_dataset, row_splits_dtype): argument
H A Dprefetching_ops.py91 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
234 def __init__(self, input_dataset, map_func, use_inter_op_parallelism=True): argument
H A Dsleep.py23 def __init__(self, input_dataset, sleep_microseconds): argument
H A Derror_ops.py58 def __init__(self, input_dataset, log_warning): argument
H A Dshuffle_ops.py29 def __init__(self, input_dataset, buffer_size, count=None, seed=None): argument
H A Dcardinality.py102 def __init__(self, input_dataset, expected_cardinality): argument
H A Dparsing_ops.py30 def __init__(self, input_dataset, features, num_parallel_calls, argument
H A Dsnapshot.py33 input_dataset, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op_test.cc230 Node* input_dataset; in TEST_F() local
287 Node* input_dataset; in TEST_F() local
337 Node* input_dataset; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/
H A Dfrom_tensor_slices_benchmark.py30 def __init__(self, input_dataset, map_func): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dtraverse_test.py32 def __init__(self, input_dataset): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dinput_ops_test.py259 def __init__(self, input_dataset): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dbatch_dataset_op_test.cc331 Node* input_dataset; in TEST() local
H A Diterator_ops.cc168 const DatasetBase* input_dataset; in Restore() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.cc837 for (auto input_dataset : input_datasets) { in RunDatasetOp() local
960 for (auto input_dataset : input_datasets) { in MakeDatasetTensor() local