Searched refs:_device_function_stack (Results 1 – 5 of 5) sorted by relevance
4672 device_fn_tmp = self._device_function_stack4673 self._device_function_stack = traceable_stack.TraceableStack()4693 self._device_function_stack = device_fn_tmp4707 self._device_function_stack.push_obj(spec, offset=total_offset)4772 old_top_of_stack = self._device_function_stack.peek_top_obj()4776 new_top_of_stack = self._device_function_stack.peek_top_obj()4779 self._device_function_stack.pop_obj()4789 for device_spec in self._device_function_stack.peek_objs():5352 def _device_function_stack(self): member in Graph5359 self._thread_local._device_function_stack = stack_copy_for_this_thread[all …]
535 old_device_stack = self._device_function_stack537 (device_stack_has_callable(graph._device_function_stack) or541 self._device_function_stack = graph._device_function_stack.copy()564 self._device_function_stack = old_device_stack
31 …self._device_function_stack = outer_graph._device_function_stack.copy() # pylint: disable=protect…
82 default_graph._device_function_stack)):
543 self._oc_dev_fn_stack = graph._device_function_stack # pylint: disable=protected-access544 …graph._device_function_stack = self._outer_device_function_stack # pylint: disable=protected-acce…552 graph._device_function_stack = self._oc_dev_fn_stack # pylint: disable=protected-access560 self._outer_device_function_stack = graph._device_function_stack.copy()