We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fd5f69d commit 9898dc7Copy full SHA for 9898dc7
stubs/tensorflow/tensorflow/config/experimental.pyi
@@ -1,3 +1,4 @@
1
+import typing_extensions
2
from _typeshed import Incomplete
3
from typing import TypedDict
4
@@ -9,6 +10,7 @@ class _MemoryInfo(TypedDict):
9
10
11
def get_memory_info(device: str) -> _MemoryInfo: ...
12
def reset_memory_stats(device: str) -> None: ...
13
+@typing_extensions.deprecated("This function is deprecated in favor of tf.config.experimental.get_memory_info")
14
def get_memory_usage(device: PhysicalDevice) -> int: ...
15
def get_memory_growth(device: PhysicalDevice) -> bool: ...
16
def set_memory_growth(device: PhysicalDevice, enable: bool) -> None: ...
0 commit comments