Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_custom_op/
H A Dautograd.py60 def mark_non_differentiable(ctx, output, output_differentiability): argument
94 output_differentiability, argument
H A Dimpl.py403 def impl_backward(self, output_differentiability=None, _stacklevel=2): argument
/aosp_15_r20/external/pytorch/torch/
H A D_custom_ops.py274 def impl_backward(qualname, output_differentiability=None, *, func=None): argument