Searched refs:container_strategy (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 19 argspec: "args=[\'self\', \'container_strategy\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 2047 def __init__(self, container_strategy): argument
2048 self._container_strategy_weakref = weakref.ref(container_strategy)
2064 container_strategy = self._container_strategy_weakref()
2065 assert container_strategy is not None
2066 return container_strategy
3409 def __init__(self, container_strategy): argument
3410 super(_DefaultDistributionExtended, self).__init__(container_strategy)
|
| D | one_device_strategy.py | 258 def __init__(self, container_strategy, device): argument
259 super(OneDeviceExtended, self).__init__(container_strategy)
|
| D | parameter_server_strategy_v2.py | 514 def __init__(self, container_strategy, cluster_resolver, argument
517 super(ParameterServerStrategyV2Extended, self).__init__(container_strategy)
|
| D | collective_all_reduce_strategy.py | 316 def __init__(self, container_strategy, cluster_resolver, argument
325 distribute_lib.StrategyExtendedV1.__init__(self, container_strategy)
327 …self._collective_key_base = container_strategy._collective_key_base # pylint: disable=protected-a…
|
| D | mirrored_strategy.py | 309 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
310 super(MirroredExtended, self).__init__(container_strategy)
336 self._collective_key_base = container_strategy._collective_key_base
|
| D | parameter_server_strategy.py | 166 container_strategy, argument
170 super(ParameterServerStrategyExtended, self).__init__(container_strategy)
|
| D | tpu_strategy.py | 787 container_strategy, argument
791 super(TPUExtended, self).__init__(container_strategy)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 36 argspec: "args=[\'self\', \'container_strategy\'], varargs=None, keywords=None, defaults=None"
|