Searched refs:container_strategy (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy.py | 70 container_strategy, argument
84 container_strategy,
|
| D | parameter_server_strategy.py | 152 def __init__(self, container_strategy, num_gpus_per_worker): argument
163 container_strategy, cluster_resolver=cluster_resolver)
|
| D | mirrored_strategy.py | 165 container_strategy, argument
175 super(MirroredExtended, self).__init__(container_strategy, devices,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 52 def __init__(self, container_strategy, device): argument
53 super(OneDeviceExtended, self).__init__(container_strategy)
|
| D | distribute_lib.py | 841 def __init__(self, container_strategy): argument
842 self._container_strategy_weakref = weakref.ref(container_strategy)
858 container_strategy = self._container_strategy_weakref()
859 assert container_strategy is not None
860 return container_strategy
|
| D | collective_all_reduce_strategy.py | 87 container_strategy, argument
91 self, container_strategy)
|
| D | parameter_server_strategy.py | 100 container_strategy, argument
102 super(ParameterServerStrategyExtended, self).__init__(container_strategy)
|
| D | tpu_strategy.py | 196 container_strategy, argument
200 super(TPUExtended, self).__init__(container_strategy)
|
| D | mirrored_strategy.py | 445 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
446 super(MirroredExtended, self).__init__(container_strategy)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 35 argspec: "args=[\'self\', \'container_strategy\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 35 argspec: "args=[\'self\', \'container_strategy\'], varargs=None, keywords=None, defaults=None"
|