Searched refs:old_variance (Results 1 – 2 of 2) sorted by relevance
148 xla::XlaOp old_variance = ctx->Input(4); in CompileImpl() local150 xla::Mul(old_variance, alpha), xla::Mul(corrected, beta)); in CompileImpl()
149 typename TTypes<U>::ConstVec old_variance(running_variance_input.vec<U>()); in operator ()() local211 one_minus_factor * old_variance + in operator ()()