Searched refs:_is_current_explicit_device (Results 1 – 1 of 1) sorted by relevance
551 def _is_current_explicit_device(device_type): function598 explicitly_on_cpu = _is_current_explicit_device('CPU')