From 0f0e261b90cc82ecaa8f527458b23bcfd6a8ccaa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ho=C3=ABl=20Bagard?=
 <34478245+hoel-bagard@users.noreply.github.com>
Date: Sat, 17 Feb 2024 12:06:56 +0900
Subject: [PATCH] `tensorflow`: Add some `tf.raw_ops` members. (#11387)

https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/raw_ops.pyi
---
 stubs/tensorflow/tensorflow/raw_ops.pyi | 45 +++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 stubs/tensorflow/tensorflow/raw_ops.pyi

diff --git a/stubs/tensorflow/tensorflow/raw_ops.pyi b/stubs/tensorflow/tensorflow/raw_ops.pyi
new file mode 100644
index 000000000..a7cf1cb71
--- /dev/null
+++ b/stubs/tensorflow/tensorflow/raw_ops.pyi
@@ -0,0 +1,45 @@
+from _typeshed import Incomplete
+from typing import Literal
+
+from tensorflow import Operation, Tensor
+from tensorflow._aliases import TensorCompatible
+
+def Fingerprint(data: TensorCompatible, method: Literal["farmhash64"], name: str | None = None) -> Tensor: ...
+def Snapshot(input: TensorCompatible, name: str | None = None) -> Tensor: ...
+def ResourceApplyAdagradV2(
+    var: Tensor,
+    accum: Tensor,
+    lr: TensorCompatible,
+    epsilon: TensorCompatible,
+    grad: TensorCompatible,
+    use_locking: bool = False,
+    update_slots: bool = True,
+    name: str | None = None,
+) -> Operation: ...
+def ResourceSparseApplyAdagradV2(
+    var: Tensor,
+    accum: Tensor,
+    lr: TensorCompatible,
+    epsilon: TensorCompatible,
+    grad: TensorCompatible,
+    indices: TensorCompatible,
+    use_locking: bool = False,
+    update_slots: bool = True,
+    name: str | None = None,
+) -> Operation: ...
+def ResourceApplyAdam(
+    var: Tensor,
+    m: Tensor,
+    v: Tensor,
+    beta1_power: TensorCompatible,
+    beta2_power: TensorCompatible,
+    lr: TensorCompatible,
+    beta1: TensorCompatible,
+    beta2: TensorCompatible,
+    epsilon: TensorCompatible,
+    grad: TensorCompatible,
+    use_locking: bool = False,
+    use_nesterov: bool = False,
+    name: str | None = None,
+) -> Operation: ...
+def __getattr__(name: str) -> Incomplete: ...