MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / assign

Method assign

tensorflow/python/distribute/values.py:1297–1302  ·  view source on GitHub ↗
(self, *args, **kwargs)

Source from the content-addressed store, hash-verified

1295 self, *args, **kwargs)
1296
1297 def assign(self, *args, **kwargs):
1298 if _enclosing_tpu_context() is None:
1299 return SyncOnReadVariable.assign(self, *args, **kwargs)
1300 else:
1301 return _make_raw_assign_fn(
1302 gen_resource_variable_ops.assign_variable_op)(self, *args, **kwargs)
1303
1304
1305def regroup(device_map, values, wrap_class=PerReplica):

Callers

nothing calls this directly

Calls 3

_make_raw_assign_fnFunction · 0.85
_enclosing_tpu_contextFunction · 0.70
assignMethod · 0.45

Tested by

no test coverage detected