Home
last modified time | relevance | path

Searched defs:tanh_softplus (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DActivation.cpp1337 const float tanh_softplus = std::tanh(std::log1p(std::exp(float(x)))); in mish_backward_kernel() local
1362 const scalar_t tanh_softplus = std::tanh(std::log1p(std::exp(x))); in mish_backward_kernel() local
1367 const Vec tanh_softplus = x_vec.exp().log1p().tanh(); in mish_backward_kernel() local
/external/pytorch/torch/csrc/autograd/
DFunctionsManual.cpp2268 const Tensor tanh_softplus = softplus.tanh(); in infinitely_differentiable_mish_backward() local