Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistributed_table_test.py116 self.assertIsInstance(lookuptable, ps_values.DistributedTable)
158 self.assertNotIsInstance(lookuptable, ps_values.DistributedTable)
201 self.assertIsInstance(lookuptable, ps_values.DistributedTable)
447 self.assertIsInstance(loaded.table, ps_values.DistributedTable)
H A Dps_values.py547 class DistributedTable(lookup_ops.StaticHashTable): class
704 class RestoredDistributedTable(DistributedTable):
H A Dparameter_server_strategy_v2.py829 return ps_values.DistributedTable(self._container_strategy(),