Home
last modified time | relevance | path

Searched defs:sign_decay (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_test.cc357 auto sign_decay = Scalar(g, 0.9); in AddSign() local
398 auto sign_decay = Scalar(g, 0.9); in PowerSign() local
H A Dtraining_ops.cc4512 const Tensor& sign_decay = ctx->input(4); in Compute() local
4619 const Tensor& sign_decay = ctx->input(4); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtraining_ops.cc999 xla::XlaOp sign_decay = ctx->Input(4); in Compile() local