Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_test.cc173 auto beta2_power = Scalar(g, 0.99); in Adam() local
Dtraining_ops.cc2837 const Tensor& beta2_power = ctx->input(4); in Compute() local
2935 T beta2_power = 0; in Compute() local
3106 const Tensor& beta2_power = ctx->input(5); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc459 xla::XlaOp beta2_power = ctx->Input(4); in Compile() local