/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
|