| /external/tensorflow/tensorflow/python/training/ |
| D | slot_creator.py | 125 with distribution_strategy.extended.colocate_vars_with(primary):
162 with distribution_strategy.extended.colocate_vars_with(primary):
|
| D | sync_replicas_optimizer.py | 264 with distribution_strategy.extended.colocate_vars_with(local_anchor):
|
| D | optimizer.py | 801 with distribution_strategy.extended.colocate_vars_with(colocate_with):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 334 def colocate_vars_with(self, colocate_with_variable): member in DistributionStrategy
336 return self._extended.colocate_vars_with(colocate_with_variable)
935 def colocate_vars_with(self, colocate_with_variable): member in DistributionStrategyExtended
1564 def colocate_vars_with(self, colocate_with_variable): member in _DefaultDistributionExtended
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy.pbtxt | 18 name: "colocate_vars_with"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 23 name: "colocate_vars_with"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 50 name: "colocate_vars_with"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy.pbtxt | 18 name: "colocate_vars_with"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 23 name: "colocate_vars_with"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 19 name: "colocate_vars_with"
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 50 name: "colocate_vars_with"
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 410 ).extended.colocate_vars_with(self.moving_mean):
418 ).extended.colocate_vars_with(self.moving_variance):
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | parameter_server_strategy_test.py | 193 with d.extended.colocate_vars_with(x):
347 with d.extended.colocate_vars_with(x):
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 542 with strategy.colocate_vars_with(var):
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2.py | 449 with self._distribution.extended.colocate_vars_with(colocate_with):
|