Searched refs:_stack_and_check_tensors (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | gradcheck.py | 710 def _stack_and_check_tensors( function
779 jacobians1, types_ok, sizes_ok = _stack_and_check_tensors(
782 jacobians2, _, _ = _stack_and_check_tensors(vjps2, inputs, output_numel)
858 jacobians1, types_ok, sizes_ok = _stack_and_check_tensors(
861 jacobians2, _, _ = _stack_and_check_tensors(vjps2, inputs, output_numel)
|