Searched refs:TPUOnReadPolicy (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 478 class TPUOnReadPolicy(values.OnReadPolicy): class
490 return super(TPUOnReadPolicy, self).assign_sub(var, *args, **kwargs)
498 return super(TPUOnReadPolicy, self).assign_add(var, *args, **kwargs)
506 return super(TPUOnReadPolicy, self).assign(var, *args, **kwargs)
|
| D | distribute_utils.py | 378 vs.VariableSynchronization.ON_READ: tpu_values_lib.TPUOnReadPolicy,
|