tensorflow: add tensorflow.audio (#11441)

python · Mar 1, 2024 · 08ddbe5 · 08ddbe5
1 parent ea2002d
commit 08ddbe5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/audio.pyi b/stubs/tensorflow/tensorflow/audio.pyi
@@ -0,0 +1,7 @@
+import tensorflow as tf
+from tensorflow._aliases import Integer, StringTensorCompatible
+
+def decode_wav(
+    contents: StringTensorCompatible, desired_channels: int = -1, desired_samples: int = -1, name: str | None = None
+) -> tuple[tf.Tensor, tf.Tensor]: ...
+def encode_wav(audio: tf.Tensor, sample_rate: Integer, name: str | None = None) -> tf.Tensor: ...