Searched refs:DistributedVariable (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values.py | 439 class DistributedVariable(DistributedDelegate): class
446 super(DistributedVariable, self).__init__(
587 ops.register_dense_tensor_like_type(DistributedVariable)
600 if not isinstance(v, DistributedVariable):
647 class MirroredVariable(DistributedVariable, Mirrored,
1228 class SyncOnReadVariable(DistributedVariable, PerReplica, trackable.Trackable):
1356 if same_id and (isinstance(v0, DistributedVariable) or
1448 not isinstance(val, DistributedVariable)):
|
| D | mirrored_strategy.py | 558 assert not isinstance(v, values.DistributedVariable)
715 assert isinstance(var, values.DistributedVariable)
|
| D | collective_all_reduce_strategy.py | 322 assert not isinstance(v, values.DistributedVariable)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | autocast_variable.py | 171 distribute_values.DistributedVariable):
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 666 distributed_values.DistributedVariable)):
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer.py | 382 if isinstance(variable, distribute_values.DistributedVariable):
|