Skip to content

Commit 9898dc7

Browse files
committed
fix: add deprecated decorator to get_memory_usage
1 parent fd5f69d commit 9898dc7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

stubs/tensorflow/tensorflow/config/experimental.pyi

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import typing_extensions
12
from _typeshed import Incomplete
23
from typing import TypedDict
34

@@ -9,6 +10,7 @@ class _MemoryInfo(TypedDict):
910

1011
def get_memory_info(device: str) -> _MemoryInfo: ...
1112
def reset_memory_stats(device: str) -> None: ...
13+
@typing_extensions.deprecated("This function is deprecated in favor of tf.config.experimental.get_memory_info")
1214
def get_memory_usage(device: PhysicalDevice) -> int: ...
1315
def get_memory_growth(device: PhysicalDevice) -> bool: ...
1416
def set_memory_growth(device: PhysicalDevice, enable: bool) -> None: ...

0 commit comments

Comments
 (0)