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

Function _assign_on_device

tensorflow/python/distribute/values.py:556–558  ·  view source on GitHub ↗
(device, variable, tensor)

Source from the content-addressed store, hash-verified

554
555
556def _assign_on_device(device, variable, tensor):
557 with ops.device(device):
558 return variable.assign(tensor)
559
560
561def _assign_add_on_device(device, variable, tensor):

Callers 2

restoreMethod · 0.85
assignMethod · 0.85

Calls 2

deviceMethod · 0.45
assignMethod · 0.45

Tested by

no test coverage detected