Home
last modified time | relevance | path

Searched refs:_prelu_kernel_backward (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py391 aten._prelu_kernel_backward,
H A Ddecompositions.py297 @register_decomposition(aten._prelu_kernel_backward)
298 def _prelu_kernel_backward( function
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesBinaryOps.cpp445 VARIADIC_BDIMS_BOXED(_prelu_kernel_backward); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DActivation.cpp734 std::tuple<Tensor, Tensor> _prelu_kernel_backward(const Tensor& grad_out, const Tensor& self, const… in _prelu_kernel_backward() function