Searched refs:device_function_stack (Results 1 – 1 of 1) sorted by relevance
5230 def _device_function_stack(self, device_function_stack): argument5233 self._thread_local._device_function_stack = device_function_stack5236 self._graph_device_function_stack = device_function_stack