Home
last modified time | relevance | path

Searched defs:negval (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DActivationLeakyReluKernel.cu30 auto negval = negval_.to<opmath_t>(); in leaky_relu_kernel() local
48 auto negval = negval_.to<opmath_t>(); in leaky_relu_backward_kernel() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dqrelu.cpp133 const Scalar& negval, Tensor& result) { in leaky_relu_out_quantized_cpu()
138 Tensor leaky_relu_quantized_cpu(const Tensor& self, const Scalar& negval) { in leaky_relu_quantized_cpu()
149 Tensor& leaky_relu_quantized_cpu_(Tensor& self, const Scalar& negval) { in leaky_relu_quantized_cpu_()
/external/pytorch/aten/src/ATen/native/cpu/
DActivation.cpp965 float negval = negval_.to<float>(); in leaky_relu_kernel() local
984 scalar_t negval = negval_.to<scalar_t>(); in leaky_relu_kernel() local
1004 float negval = negval_.to<float>(); in leaky_relu_backward_kernel() local
1024 scalar_t negval = negval_.to<scalar_t>(); in leaky_relu_backward_kernel() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp623 float negval = negval_.to<float>(); in leaky_qrelu_out_kernel() local