Searched defs:exp_grad (Results 1 – 1 of 1) sorted by relevance
68 const auto exp_grad = xla::Mul(grad, xla::Add(activation, one)); in Compile() local101 const auto exp_grad = xla::Mul(grad, xla::Add(activation, scale_alpha)); in Compile() local