Searched refs:input_contexts (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 113 input_contexts, argument
162 input_contexts, strategy, options)
167 input_contexts,
1233 def __init__(self, dataset_fn, input_workers, input_contexts, strategy, argument
1251 if input_workers.num_workers != len(input_contexts):
1255 (input_workers.num_workers, len(input_contexts)))
1258 self._input_contexts = input_contexts
1375 def __init__(self, input_fn, input_workers, input_contexts, strategy): argument
1393 if input_workers.num_workers != len(input_contexts):
1397 (input_workers.num_workers, len(input_contexts)))
[all …]
|
| D | mirrored_strategy.py | 548 input_contexts = []
551 input_contexts.append(distribute_lib.InputContext(
556 input_contexts,
579 input_contexts = []
582 input_contexts.append(distribute_lib.InputContext(
588 dataset_fn, input_workers, input_contexts, self._container_strategy(),
|
| D | tpu_strategy.py | 891 input_contexts = []
896 input_contexts.append(distribute_lib.InputContext(
903 input_contexts,
960 input_contexts = []
963 input_contexts.append(distribute_lib.InputContext(
971 input_contexts,
|
| D | input_lib_test.py | 81 input_contexts = []
83 input_contexts.append(
94 input_contexts,
|
| D | collective_all_reduce_strategy.py | 621 input_contexts=[input_context],
|