-
Notifications
You must be signed in to change notification settings - Fork 0
/
raw_ops.pyi
43 lines (41 loc) · 1.17 KB
/
raw_ops.pyi
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from typing import Literal
import tensorflow as tf
from tensorflow import 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: tf.Tensor,
accum: tf.Tensor,
lr: TensorCompatible,
epsilon: TensorCompatible,
grad: TensorCompatible,
use_locking: bool = False,
update_slots: bool = True,
) -> None: ...
def ResourceSparseApplyAdagradV2(
var: tf.Tensor,
accum: tf.Tensor,
lr: TensorCompatible,
epsilon: TensorCompatible,
grad: TensorCompatible,
indices: TensorCompatible,
use_locking: bool = False,
update_slots: bool = True,
) -> None: ...
def ResourceApplyAdam(
var: tf.Tensor,
m: tf.Tensor,
v: tf.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,
) -> None: ...