Searched refs:CentralStorageStrategy (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | central_storage_strategy.py | 28 class CentralStorageStrategy(distribute_lib.Strategy): class
69 super(CentralStorageStrategy, self).__init__(extended)
112 return super(CentralStorageStrategy, self).experimental_distribute_dataset(
129 return super(CentralStorageStrategy, self).experimental_local_results(value)
147 return super(CentralStorageStrategy, self).run(fn, args, kwargs, options)
213 return super(CentralStorageStrategy, self).reduce(reduce_op, value, axis)
219 __doc__ = CentralStorageStrategy.__doc__
230 __init__.__doc__ = CentralStorageStrategy.__init__.__doc__
|
| D | strategy_combinations.py | 61 CentralStorageStrategy = _version_chooser( variable
63 central_storage_strategy.CentralStorageStrategy)
278 lambda: CentralStorageStrategy(["/gpu:0", "/gpu:1"]),
282 lambda: CentralStorageStrategy(["/gpu:0", "/cpu:0"]),
|
| D | strategy_gather_test.py | 199 central_storage_strategy.CentralStorageStrategy)) and pure_eager:
251 central_storage_strategy.CentralStorageStrategy)):
517 central_storage_strategy.CentralStorageStrategy)) and pure_eager:
573 central_storage_strategy.CentralStorageStrategy)):
|
| D | strategy_combinations_test.py | 186 central_storage_strategy.CentralStorageStrategy)
|
| D | parameter_server_strategy_test.py | 92 central_storage_strategy.CentralStorageStrategy._from_num_gpus(num_gpus)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | utils.py | 95 (central_storage_strategy.CentralStorageStrategy,
116 (central_storage_strategy.CentralStorageStrategy,
137 (central_storage_strategy.CentralStorageStrategy,
|
| D | optimizer_v2.py | 657 central_storage_strategy.CentralStorageStrategy,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 1 path: "tensorflow.distribute.experimental.CentralStorageStrategy"
3 …nstance: "<class \'tensorflow.python.distribute.central_storage_strategy.CentralStorageStrategy\'>"
|
| D | tensorflow.distribute.experimental.pbtxt | 4 name: "CentralStorageStrategy"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.experimental.pbtxt | 4 name: "CentralStorageStrategy"
|
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 1 path: "tensorflow.distribute.experimental.CentralStorageStrategy"
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_rnn_model_correctness_test.py | 121 (central_storage_strategy.CentralStorageStrategy,
|
| D | distribute_strategy_test.py | 510 central_storage_strategy.CentralStorageStrategy,
562 central_storage_strategy.CentralStorageStrategy,
1995 (central_storage_strategy.CentralStorageStrategy,
2008 (central_storage_strategy.CentralStorageStrategy,
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer_test.py | 956 strategy = central_storage_strategy.CentralStorageStrategy()
|
| D | keras_test.py | 114 return central_storage_strategy.CentralStorageStrategy(
|
| /external/tensorflow/ |
| D | RELEASE.md | 27 …y` scopes is now allowed (except for `tf.distribute.experimental.CentralStorageStrategy` and `tf.d…
|