Home
last modified time | relevance | path

Searched refs:_is_float_or_complex_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgradcheck.py46 def _is_float_or_complex_tensor(obj): function
60 if _is_float_or_complex_tensor(t) and t.requires_grad:
75 if _is_float_or_complex_tensor(t):
553 dual_outputs = filter(_is_float_or_complex_tensor, raw_outputs)
578 dual_outputs = filter(_is_float_or_complex_tensor, raw_outputs)
686 if _is_float_or_complex_tensor(output):
1091 if not _is_float_or_complex_tensor(current_input):
1263 dual_outputs1 = filter(_is_float_or_complex_tensor, raw_outputs)
1268 dual_outputs2 = filter(_is_float_or_complex_tensor, raw_outputs)