| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 4388 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 D | readers.py | 321 input_dataset, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | testing.py | 121 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 D | distribute.py | 70 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 D | grouping.py | 263 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 D | batching.py | 304 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 D | prefetching_ops.py | 91 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 D | sleep.py | 23 def __init__(self, input_dataset, sleep_microseconds): argument
|
| H A D | error_ops.py | 58 def __init__(self, input_dataset, log_warning): argument
|
| H A D | shuffle_ops.py | 29 def __init__(self, input_dataset, buffer_size, count=None, seed=None): argument
|
| H A D | cardinality.py | 102 def __init__(self, input_dataset, expected_cardinality): argument
|
| H A D | parsing_ops.py | 30 def __init__(self, input_dataset, features, num_parallel_calls, argument
|
| H A D | snapshot.py | 33 input_dataset, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | auto_shard_dataset_op_test.cc | 230 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 D | from_tensor_slices_benchmark.py | 30 def __init__(self, input_dataset, map_func): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | traverse_test.py | 32 def __init__(self, input_dataset): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | input_ops_test.py | 259 def __init__(self, input_dataset): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | batch_dataset_op_test.cc | 331 Node* input_dataset; in TEST() local
|
| H A D | iterator_ops.cc | 168 const DatasetBase* input_dataset; in Restore() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.cc | 837 for (auto input_dataset : input_datasets) { in RunDatasetOp() local
960 for (auto input_dataset : input_datasets) { in MakeDatasetTensor() local
|