Searched refs:_device_function_stack (Results 1 – 4 of 4) sorted by relevance
387 old_device_stack = self._device_function_stack390 self._device_function_stack = self._device_function_stack.copy()394 device_stack_has_callable(graph._device_function_stack)):396 self._device_function_stack = graph._device_function_stack.copy()423 self._device_function_stack = old_device_stack
4184 device_fn_tmp = self._device_function_stack4185 self._device_function_stack = traceable_stack.TraceableStack()4201 self._device_function_stack = device_fn_tmp4213 self._device_function_stack.push_obj(spec, offset=total_offset)4278 old_top_of_stack = self._device_function_stack.peek_top_obj()4282 new_top_of_stack = self._device_function_stack.peek_top_obj()4285 self._device_function_stack.pop_obj()4295 for device_spec in self._device_function_stack.peek_objs():4847 def _device_function_stack(self): member in Graph4854 self._thread_local._device_function_stack = stack_copy_for_this_thread[all …]
466 self._oc_dev_fn_stack = graph._device_function_stack # pylint: disable=protected-access467 …graph._device_function_stack = self._outer_device_function_stack # pylint: disable=protected-acce…475 graph._device_function_stack = self._oc_dev_fn_stack # pylint: disable=protected-access483 self._outer_device_function_stack = graph._device_function_stack.copy()
2615 default_graph._device_function_stack)):