Home
last modified time | relevance | path

Searched refs:device_stack (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py92 device_stack = ops.get_default_graph()._device_functions_outer_to_inner
94 return not bool(device_stack)
/external/tensorflow/tensorflow/python/eager/
Dcontext.py247 device_stack=None)
249 def push(self, is_building_function, enter_context_fn, device_stack): argument
267 ContextSwitch(is_building_function, enter_context_fn, device_stack))
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py1328 def device_stack_has_callable(device_stack): argument
1332 for spec in device_stack.peek_objs())
Dops.py5904 innermost_nonempty_device_stack = stack_entry.device_stack