Searched refs:same_two_models (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/repro/ |
| H A D | after_dynamo.py | 29 same_two_models,
439 assert same_two_models(mod, mod, args), "Eager itself failed"
440 if not same_two_models(
|
| H A D | after_aot.py | 35 same_two_models,
143 failed = not same_two_models(
722 if not same_two_models(
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | debug_utils.py | 336 def same_two_models( function
450 return not same_two_models(
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cuda_repro.py | 12 from torch._dynamo.debug_utils import same_two_models
372 assert same_two_models(mod, opt_mod, args), "Dynamo failed"
655 assert same_two_models(
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_modules.py | 21 from torch._dynamo.debug_utils import same_two_models
2948 self.assertTrue(same_two_models(loaded_model, mod, [inp]))
2949 self.assertTrue(same_two_models(loaded_model, opt_mod, [inp]))
|
| H A D | test_repros.py | 38 from torch._dynamo.debug_utils import same_two_models
2478 self.assertTrue(same_two_models(mod, opt_mod, args))
2493 self.assertTrue(same_two_models(mod, opt_mod, args))
2787 self.assertTrue(same_two_models(mod, opt_mod, args))
|