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.py83 packed_var = var._packed_variable # pylint: disable=protected-access
168 if var._packed_variable is not None: # pylint: disable=protected-access
171 …assign_func(d, var._packed_variable, value) for d in var._devices)) # pylint: disable=protected-a…
Dtpu_strategy.py1296 values.DistributedVariable) and value._packed_variable is not None:
1298 value._packed_variable.on_device(d)
1299 for d in value._packed_variable.devices)
1361 values.DistributedVariable) and value._packed_variable is not None:
1363 value._packed_variable.on_device(d)
1364 for d in value._packed_variable.devices)
1395 packed_var = var._packed_variable # pylint: disable=protected-access
Ddistributed_variable_test.py607 self.assertIsInstance(v._packed_variable,
612 self.assertIsNone(v._packed_variable)
Dvalues.py650 def _packed_variable(self): member in DistributedVariable