Searched refs:container_strategy (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_function_strategy.py | 84 def __init__(self, container_strategy, devices): argument
85 super(MirroredFunctionExtended, self).__init__(container_strategy)
|
| D | distribute_lib.py | 1349 def __init__(self, container_strategy): argument
1350 self._container_strategy_weakref = weakref.ref(container_strategy)
1366 container_strategy = self._container_strategy_weakref()
1367 assert container_strategy is not None
1368 return container_strategy
2180 def __init__(self, container_strategy): argument
2181 super(_DefaultDistributionExtended, self).__init__(container_strategy)
|
| D | one_device_strategy.py | 248 def __init__(self, container_strategy, device): argument
249 super(OneDeviceExtended, self).__init__(container_strategy)
|
| D | collective_all_reduce_strategy.py | 165 container_strategy, argument
169 distribute_lib.StrategyExtendedV1.__init__(self, container_strategy)
|
| D | parameter_server_strategy.py | 147 container_strategy, argument
151 super(ParameterServerStrategyExtended, self).__init__(container_strategy)
|
| D | tpu_strategy.py | 221 container_strategy, argument
225 super(TPUExtended, self).__init__(container_strategy)
|
| D | mirrored_strategy.py | 451 def __init__(self, container_strategy, devices=None, cross_device_ops=None): argument
452 super(MirroredExtended, self).__init__(container_strategy)
|
| /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/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 36 argspec: "args=[\'self\', \'container_strategy\'], varargs=None, keywords=None, defaults=None"
|