Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DActivationLeakyReluKernel.cu22 void leaky_relu_kernel(TensorIteratorBase& iter, const Scalar& negval_) { in leaky_relu_kernel()
40 const Scalar& negval_) { in leaky_relu_backward_kernel()
/external/pytorch/aten/src/ATen/native/cpu/
DActivation.cpp960 static void leaky_relu_kernel(TensorIteratorBase& iter, const Scalar& negval_) { in leaky_relu_kernel()
999 static void leaky_relu_backward_kernel(TensorIteratorBase& iter, const Scalar& negval_) { in leaky_relu_backward_kernel()
/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp613 const Scalar& negval_) { in leaky_qrelu_out_kernel()