Home
last modified time | relevance | path

Searched refs:_packed_variable (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dvalues_util.py87 packed_var = var._packed_variable # pylint: disable=protected-access
172 if var._packed_variable is not None: # pylint: disable=protected-access
175 …assign_func(d, var._packed_variable, value) for d in var._devices)) # pylint: disable=protected-a…
Dtpu_strategy.py1176 values.DistributedVariable) and value._packed_variable is not None:
1178 value._packed_variable.on_device(d)
1179 for d in value._packed_variable.devices)
1239 values.DistributedVariable) and value._packed_variable is not None:
1241 value._packed_variable.on_device(d)
1242 for d in value._packed_variable.devices)
1275 packed_var = var._packed_variable # pylint: disable=protected-access
Dvalues_test.py931 v._packed_variable, packed.PackedDistributedVariable)
935 self.assertIsNone(v._packed_variable)
Dvalues.py613 def _packed_variable(self): member in DistributedVariable