Searched defs:jacobian_input (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_nn.py | 3247 …def _analytical_jacobian(self, module, input: _TensorOrTensors, jacobian_input=True, jacobian_para… argument
3289 …def _numerical_jacobian(self, module, input: _TensorOrTensors, jacobian_input=True, jacobian_param… argument
3306 def check_jacobian(self, module, input: _TensorOrTensors, jacobian_input=True): argument
|