Home
last modified time | relevance | path

Searched defs:negcoef (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DActivationEluKernel.cu34 auto negcoef = alpha.to<opmath_t>() * scale.to<opmath_t>(); in elu_kernel() local
60 auto negcoef = alpha.to<opmath_t>() * scale.to<opmath_t>(); in elu_backward_kernel() local
/external/pytorch/aten/src/ATen/native/cpu/
DActivation.cpp192 auto negcoef = alpha.to<float>() * scale.to<float>(); in elu_kernel() local
221 auto negcoef = alpha.to<scalar_t>() * scale.to<scalar_t>(); in elu_kernel() local
249 auto negcoef = alpha.to<float>() * scale.to<float>(); in elu_backward_kernel() local
286 auto negcoef = alpha.to<scalar_t>() * scale.to<scalar_t>(); in elu_backward_kernel() local
/external/pytorch/torch/csrc/jit/tensorexpr/
Dlowerings.cpp1353 auto negcoef = Cast::make(a.dtype(), alpha) * poscoef; in nnc_lowerings_lazy_registration() local