Home
last modified time | relevance | path

Searched refs:replica_local_var (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py164 def read_var(self, replica_local_var): argument
166 return array_ops.identity(replica_local_var)
Dmirrored_strategy.py737 def read_var(self, replica_local_var): argument
739 if isinstance(replica_local_var, values.SyncOnReadVariable):
740 return replica_local_var._get_cross_replica() # pylint: disable=protected-access
741 assert isinstance(replica_local_var, values.Mirrored)
742 return array_ops.identity(replica_local_var.get())
Ddistribute_lib.py1625 def read_var(self, replica_local_var): argument
1626 return array_ops.identity(replica_local_var)