add tf.distribute.experimental.coordinator.RemoteValue

python · Feb 4, 2024 · b6ae7c9 · b6ae7c9
1 parent 7a6a749
commit b6ae7c9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi b/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi
@@ -0,0 +1,9 @@
+from typing import Generic, TypeVar
+
+from tensorflow._aliases import AnyArray
+
+_Value = TypeVar("_Value", covariant=True)
+
+class RemoteValue(Generic[_Value]):
+    def fetch(self) -> AnyArray: ...
+    def get(self) -> _Value: ...