Home
last modified time | relevance | path

Searched refs:optimizer_parameter_map (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Doptim_baseline.py106 def emit(optimizer_parameter_map): argument
110 for optimizer_name, parameters in optimizer_parameter_map.items():
132 optimizer_parameter_map = {}
135 optimizer_parameter_map[optimizer] = run(
139 emit(optimizer_parameter_map)