Home
last modified time | relevance | path

Searched defs:is_differentiable (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dcustom_function.cpp118 bool is_differentiable = in _process_forward_mode_AD() local
280 bool is_saved_and_setup_context) { in _process_backward_mode_ad()
371 bool is_differentiable = cdata && in _process_backward_mode_ad() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dop_selector.py21 def is_differentiable(op): function
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dautograd.py827 def is_differentiable( function
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ops.py2438 def is_differentiable(inp): function