Searched refs:features_exp (Results 1 – 3 of 3) sorted by relevance
52 auto features_exp = features.exp(); in operator() local55 too_small.select(features_exp, // softplus(x) ~= exp(x) for x small in operator()56 features_exp.log1p())); in operator()
106 xla::XlaOp features_exp = Exp(features); in Softplus() local109 Select(too_small, features_exp, Log1p(features_exp))); in Softplus()
702 (HLO_ExpOp:$features_exp $features),725 $features_exp,726 (HLO_Log1pOp $features_exp)