From 4fa759f12a08a5be85734c413ddb78027848e76a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ho=C3=ABl=20Bagard?= <34478245+hoel-bagard@users.noreply.github.com> Date: Fri, 26 Jan 2024 09:48:54 +0900 Subject: [PATCH] Add `convert_to_tensor` to `tensorflow` (#11292) --- stubs/tensorflow/tensorflow/__init__.pyi | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi index 63b37b3d8..18f49fe37 100644 --- a/stubs/tensorflow/tensorflow/__init__.pyi +++ b/stubs/tensorflow/tensorflow/__init__.pyi @@ -406,3 +406,9 @@ class RaggedTensorSpec(TypeSpec[struct_pb2.TypeSpecProto]): def from_value(cls, value: RaggedTensor) -> Self: ... def __getattr__(name: str) -> Incomplete: ... +def convert_to_tensor( + value: _TensorCompatible | IndexedSlices, + dtype: _DTypeLike | None = None, + dtype_hint: _DTypeLike | None = None, + name: str | None = None, +) -> Tensor: ...