Searched refs:_assign_func (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values.py | 663 def _assign_func(self, *args, **kwargs): member in MirroredVariable
700 return self._assign_func(f=assign_sub_fn, *args, **kwargs)
704 return self._assign_func(f=assign_add_fn, *args, **kwargs)
708 return self._assign_func(f=assign_fn, *args, **kwargs)
933 def _assign_func(self, *args, **kwargs): member in TPUMirroredVariable
1014 return self._assign_func(f=assign_sub_fn, *args, **kwargs)
1029 return self._assign_func(f=assign_add_fn, *args, **kwargs)
1044 return self._assign_func(f=assign_fn, *args, **kwargs)
1477 def _assign_func(self, *args, **kwargs): member in AggregatingVariable
1510 return self._assign_func(f=assign_sub_fn, *args, **kwargs)
[all …]
|