Searched refs:device_function_stack (Results 1 – 1 of 1) sorted by relevance
5389 def _device_function_stack(self, device_function_stack): argument5392 self._thread_local._device_function_stack = device_function_stack5395 self._graph_device_function_stack = device_function_stack