Searched defs:sig_b (Results 1 – 2 of 2) sorted by relevance
130 const auto sig_b = at::sigmoid(b); in glu_backward_jvp() local
1173 const auto sig_b = one / (one + std::exp(-b)); in glu_jvp_kernel() local1177 const auto sig_b = ones / (ones + b.neg().exp()); in glu_jvp_kernel() local