Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/DeepRec-AI/DeepRec
/ _assign_sub_on_device
Function
_assign_sub_on_device
tensorflow/python/distribute/values.py:566–568 ·
view source on GitHub ↗
(device, variable, tensor)
Source
from the content-addressed store, hash-verified
564
565
566
def
_assign_sub_on_device(device, variable, tensor):
567
with
ops.device(device):
568
return
variable.assign_sub(tensor)
569
570
571
def
_assert_strategy(strategy):
Callers
1
assign_sub
Method · 0.85
Calls
2
device
Method · 0.45
assign_sub
Method · 0.45
Tested by
no test coverage detected