Searched refs:_device_function_stack (Results 1 – 5 of 5) sorted by relevance
35 …self._device_function_stack = outer_graph._device_function_stack.copy() # pylint: disable=protect…
4413 device_fn_tmp = self._device_function_stack4414 self._device_function_stack = traceable_stack.TraceableStack()4434 self._device_function_stack = device_fn_tmp4448 self._device_function_stack.push_obj(spec, offset=total_offset)4513 old_top_of_stack = self._device_function_stack.peek_top_obj()4517 new_top_of_stack = self._device_function_stack.peek_top_obj()4520 self._device_function_stack.pop_obj()4530 for device_spec in self._device_function_stack.peek_objs():5085 def _device_function_stack(self): member in Graph5092 self._thread_local._device_function_stack = stack_copy_for_this_thread[all …]
393 old_device_stack = self._device_function_stack395 (device_stack_has_callable(graph._device_function_stack) or399 self._device_function_stack = graph._device_function_stack.copy()422 self._device_function_stack = old_device_stack
519 self._oc_dev_fn_stack = graph._device_function_stack # pylint: disable=protected-access520 …graph._device_function_stack = self._outer_device_function_stack # pylint: disable=protected-acce…528 graph._device_function_stack = self._oc_dev_fn_stack # pylint: disable=protected-access536 self._outer_device_function_stack = graph._device_function_stack.copy()
3216 default_graph._device_function_stack)):