Home
last modified time | relevance | path

Searched defs:threshold_backward (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DActivation.cpp100 TORCH_META_FUNC(threshold_backward)(const Tensor& grad, const Tensor& self, const Scalar& threshold… in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py222 def threshold_backward(grad_output: Tensor, self: Tensor, threshold: float): function