Searched refs:alpha_side_input (Results 1 – 1 of 1) sorted by relevance
43 HloConstantInstruction* alpha_side_input; member204 double alpha_side_input; in TryRewriteToCudnnForwardRelu() local206 if (match.alpha_side_input) { in TryRewriteToCudnnForwardRelu()207 TF_ASSIGN_OR_RETURN(alpha_side_input, in TryRewriteToCudnnForwardRelu()208 get_alpha_value(match.alpha_side_input)); in TryRewriteToCudnnForwardRelu()210 alpha_side_input = 1; in TryRewriteToCudnnForwardRelu()213 CHECK(match.alpha_side_input == nullptr); in TryRewriteToCudnnForwardRelu()214 alpha_side_input = 0; in TryRewriteToCudnnForwardRelu()250 config.set_side_input_scale(alpha_side_input); in TryRewriteToCudnnForwardRelu()