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.cc258 auto beta2_power = Scalar(g, 0.99); in Adam() local
Dtraining_ops.cc3545 const Tensor& beta2_power = ctx->input(4); in Compute() local
3699 const Tensor& beta2_power = ctx->input(5); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc523 xla::XlaOp beta2_power = ctx->Input(4); in Compile() local