Searched defs:beta1_power (Results 1 – 3 of 3) sorted by relevance
522 xla::XlaOp beta1_power = ctx->Input(3); in Compile() local613 xla::XlaOp beta1_power = ctx->Input(3); in Compile() local
258 auto beta1_power = Scalar(g, 0.9); in Adam() local
3552 const Tensor& beta1_power = ctx->input(3); in Compute() local3698 const Tensor& beta1_power = ctx->input(4); in Compute() local3833 const Tensor& beta1_power = ctx->input(3); in Compute() local