Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DActivationEluKernel.cu35 auto poscoef = scale.to<opmath_t>(); in elu_kernel() local
61 auto poscoef = scale.to<opmath_t>(); in elu_backward_kernel() local
/external/pytorch/aten/src/ATen/native/cpu/
DActivation.cpp193 auto poscoef = scale.to<float>(); in elu_kernel() local
222 auto poscoef = scale.to<scalar_t>(); in elu_kernel() local
250 auto poscoef = scale.to<float>(); in elu_backward_kernel() local
287 auto poscoef = scale.to<scalar_t>(); in elu_backward_kernel() local
/external/pytorch/torch/csrc/jit/tensorexpr/
Dlowerings.cpp1351 auto poscoef = Cast::make(a.dtype(), scale); in nnc_lowerings_lazy_registration() local