(raw_assign_fn)
| 797 | |
| 798 | |
| 799 | def _make_raw_assign_fn(raw_assign_fn): # pylint: disable=missing-docstring |
| 800 | def assign_fn(var, value, use_locking=False, name=None, read_value=True): # pylint: disable=missing-docstring |
| 801 | del use_locking # Unused. |
| 802 | |
| 803 | with _maybe_enter_graph(var.handle): |
| 804 | op = raw_assign_fn( |
| 805 | var.handle, ops.convert_to_tensor(value, dtype=var.dtype), name=name) |
| 806 | |
| 807 | with ops.control_dependencies([op]): |
| 808 | return var._read_variable_op() if read_value else op # pylint: disable=protected-access |
| 809 | return assign_fn |
| 810 | |
| 811 | |
| 812 | class TPUVariableMixin(object): |
no outgoing calls
no test coverage detected