Searched refs:device_function_stack (Results 1 – 1 of 1) sorted by relevance
4989 def _device_function_stack(self, device_function_stack): argument4992 self._thread_local._device_function_stack = device_function_stack4995 self._graph_device_function_stack = device_function_stack