Searched refs:initialization_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | dataloader.cpp | 101 auto initialization_function = [&](size_t preloader_count, in TEST() local
124 initialization_function(0, 1, 1), in TEST()
128 initialization_function(1, 0, 1), in TEST()
132 initialization_function(1, 1, 0), in TEST()
136 initialization_function(1, 10, 5), in TEST()
140 initialization_function(1, 10, 20, 0), in TEST()
|