Searched refs:test_serialize_optimizer (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | serialize.cpp | 77 void test_serialize_optimizer( in test_serialize_optimizer() function
540 test_serialize_optimizer<Adagrad, AdagradOptions, AdagradParamState>( in TEST()
584 test_serialize_optimizer<SGD, SGDOptions, SGDParamState>( in TEST()
633 test_serialize_optimizer<Adam, AdamOptions, AdamParamState>( in TEST()
691 test_serialize_optimizer<AdamW, AdamWOptions, AdamWParamState>( in TEST()
751 test_serialize_optimizer<RMSprop, RMSpropOptions, RMSpropParamState>(options); in TEST()
820 test_serialize_optimizer<LBFGS, LBFGSOptions, LBFGSParamState>( in TEST()
|