Searched defs:input_workers (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | input_lib.py | 494 def _calculate_replicas_with_values(strategy, input_workers, optional_list): argument
556 def __init__(self, input_workers, iterators, strategy, cardinality, argument
675 input_workers, argument
862 input_workers=None, argument
924 def __init__(self, input_workers): argument
1000 input_workers, argument
1119 input_workers, strategy, argument
1318 input_workers, argument
1530 def _get_value_or_dummy(input_workers, optional_list, produce_dummy): argument
1847 input_workers, argument
[all …]
|
| H A D | input_util.py | 23 input_workers, argument
79 input_workers, argument
|
| H A D | input_lib_test.py | 67 input_workers, argument
106 input_workers, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | input_lib.py | 33 input_workers, argument
232 input_workers, argument
277 def __init__(self, input_fn, input_workers, input_contexts, strategy): argument
405 def _create_iterators_per_worker(worker_datasets, input_workers, options=None): argument
|