Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dinit_baseline.py28 def emit(initializer_parameter_map): argument
32 for initializer_name, weights in initializer_parameter_map.items():
66 initializer_parameter_map = {}
69 initializer_parameter_map[initializer] = run(initializer)
71 emit(initializer_parameter_map)