Searched defs:beta1_power (Results 1 – 3 of 3) sorted by relevance
458 xla::XlaOp beta1_power = ctx->Input(3); in Compile() local538 xla::XlaOp beta1_power = ctx->Input(3); in Compile() local
172 auto beta1_power = Scalar(g, 0.9); in Adam() local
2836 const Tensor& beta1_power = ctx->input(3); in Compute() local2934 T beta1_power = 0; in Compute() local3105 const Tensor& beta1_power = ctx->input(4); in Compute() local3240 const Tensor& beta1_power = ctx->input(3); in Compute() local