Searched refs:make_recompile_test (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_optimizers.py | 442 def make_recompile_test(optim_cls, closure=None, kernel_count=2, **kwargs): function
620 test_adam_recompile = make_recompile_test(Adam, lr=0.01)
621 test_adamw_recompile = make_recompile_test(AdamW, lr=0.01)
622 test_adamax_recompile = make_recompile_test(Adamax, lr=0.01)
623 test_nadam_recompile = make_recompile_test(NAdam, lr=0.01)
624 test_rprop_recompile = make_recompile_test(Rprop, lr=0.01, kernel_count=2)
625 test_rmsprop_recompile = make_recompile_test(RMSprop, lr=0.01)
626 test_adadelta_recompile = make_recompile_test(Adadelta, lr=0.01)
627 test_adagrad_recompile = make_recompile_test(Adagrad, lr=0.01)
628 test_asgd_recompile_default = make_recompile_test(ASGD, lr=0.01)
[all …]
|