Searched defs:mark_non_differentiable (Results 1 – 3 of 3) sorted by relevance
60 def mark_non_differentiable(ctx, output, output_differentiability): function
194 def mark_non_differentiable(self, *args: torch.Tensor): member in FunctionCtx828 def mark_non_differentiable(self, *args: Any, **kwargs: Any) -> None: member in NestedIOFunction
561 void AutogradContext::mark_non_differentiable(const variable_list& outputs) { in mark_non_differentiable() function in torch::autograd::AutogradContext