fix: reorganize config module.

Author: hoel-bagard

stubs/tensorflow/tensorflow/config.pyi renamed to stubs/tensorflow/tensorflow/config/__init__.pyi

@@ -1,9 +1,10 @@
 from typing import NamedTuple
 
+from tensorflow.config import experimental as experimental
+
 class PhysicalDevice(NamedTuple):
     name: str
     device_type: str
 
 def list_physical_devices(device_type: None | str = None) -> list[PhysicalDevice]: ...
-def set_memory_growth(device: PhysicalDevice, enable: bool) -> None: ...
 def set_visible_devices(devices: list[PhysicalDevice], device_type: None | str = None) -> None: ...

stubs/tensorflow/tensorflow/config/experimental.pyi

@@ -0,0 +1,3 @@
+from tensorflow.config import PhysicalDevice
+
+def set_memory_growth(device: PhysicalDevice, enable: bool) -> None: ...