Searched refs:verify_aot_autograd (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_aotdispatch.py | 352 def verify_aot_autograd( member in TestAOTAutograd
481 self.verify_aot_autograd(f, inp)
483 self.verify_aot_autograd(f, inp)
490 self.verify_aot_autograd(f, inp)
492 self.verify_aot_autograd(f, inp)
499 self.verify_aot_autograd(f, inp)
501 self.verify_aot_autograd(f, inp)
508 self.verify_aot_autograd(f, inp)
510 self.verify_aot_autograd(f, inp)
520 self.verify_aot_autograd(f, inp, dynamic=True)
[all …]
|