Home
last modified time | relevance | path

Searched refs:set_cpu0 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dsaver.py149 device = saveable_object_util.set_cpu0(saveable.device)
264 with ops.device(saveable_object_util.set_cpu0(device)):
272 with ops.device(saveable_object_util.set_cpu0(last_device)):
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py41 def set_cpu0(device_string): function