Searched refs:device_function_stack (Results 1 – 1 of 1) sorted by relevance
5122 def _device_function_stack(self, device_function_stack): argument5125 self._thread_local._device_function_stack = device_function_stack5128 self._graph_device_function_stack = device_function_stack