Searched refs:_check_device (Results 1 – 1 of 1) sorted by relevance
67 _check_device(grad, expected=op.device)165 _check_device(grad, expected=op.device)183 _check_device(tensor)204 _check_device(t)221 _check_device(t)245 _check_device(t)263 def _check_device(tensor, expected=None): function