Searched defs:non_differentiable (Results 1 – 5 of 5) sorted by relevance
33 const std::unordered_set<at::TensorImpl*>& non_differentiable, in _process_forward_mode_AD()258 const std::unordered_set<at::TensorImpl*>& non_differentiable, in _process_backward_mode_ad()440 const std::unordered_set<at::TensorImpl*>& non_differentiable, in _wrap_outputs()
107 PyObject* non_differentiable; member
618 auto non_differentiable = _parse_non_differentiable(self); in _wrap_outputs() local
38 non_differentiable = collections.namedtuple('non_differentiable', ['tensor']) variable
832 non_differentiable=None, argument