tensorflow: Add some tf.raw_ops members. (#11387)

https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/raw_ops.pyi
This commit is contained in:
Hoël Bagard
2024-02-17 12:06:56 +09:00
committed by GitHub
parent aed633ec95
commit 0f0e261b90

View File

@@ -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: ...