diff --git a/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi b/stubs/tensorflow/tensorflow/distribute/experimental/coordinator.pyi new file mode 100644 index 000000000000..8c8ca3139669 --- /dev/null +++ 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: ...