Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py787 def read_var(self, replica_local_var): argument
790 if distribute_utils.is_sync_on_read(replica_local_var):
791 return replica_local_var._get_cross_replica()
792 assert distribute_utils.is_mirrored(replica_local_var)
793 return array_ops.identity(replica_local_var._get())
Done_device_strategy.py418 def read_var(self, replica_local_var): argument
420 return array_ops.identity(replica_local_var)
Ddistribute_lib.py3489 def read_var(self, replica_local_var): argument
3490 return array_ops.identity(replica_local_var)