Searched refs:cross_layer_equalization (Results 1 – 2 of 2) sorted by relevance
116 def cross_layer_equalization(module1, module2, output_axis=0, input_axis=1): function252 cross_layer_equalization(name_to_module[pair[0]], name_to_module[pair[1]])
44 _equalize.cross_layer_equalization(module1, module2)