Searched refs:TPUVariableMixin (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 76 class TPUVariableMixin(object): class
80 super(TPUVariableMixin, self).__init__(*args, **kwargs)
91 return super(TPUVariableMixin, self).__getattr__(name)
98 return super(TPUVariableMixin, self).get()
150 return super(TPUVariableMixin, self).read_value()
156 return super(TPUVariableMixin, self).value()
162 return super(TPUVariableMixin, self)._as_graph_element() # pylint: disable=protected-access
179 return super(TPUVariableMixin, self)._dense_var_to_tensor(
188 class TPUDistributedVariable(TPUVariableMixin, values.DistributedVariable):
255 class TPUMirroredVariable(TPUVariableMixin, values.MirroredVariable):
[all …]
|
| D | tpu_strategy.py | 1263 assert isinstance(var, tpu_values.TPUVariableMixin) or isinstance(
1301 assert isinstance(var, tpu_values.TPUVariableMixin) or isinstance(
|