Searched refs:selected_optimized_path (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | op_mul.cpp | 76 const ElementwiseOptimizedPath selected_optimized_path) { in handle_last_dim_broadcast() argument
80 if (selected_optimized_path == in handle_last_dim_broadcast()
115 const ElementwiseOptimizedPath selected_optimized_path) { in handle_broadcast_mul() argument
116 if ((selected_optimized_path == in handle_broadcast_mul()
118 (selected_optimized_path == in handle_broadcast_mul()
120 return handle_last_dim_broadcast(ctx, a, b, out, selected_optimized_path); in handle_broadcast_mul()
126 if ((selected_optimized_path == in handle_broadcast_mul()
128 (selected_optimized_path == in handle_broadcast_mul()
135 (selected_optimized_path == in handle_broadcast_mul()
137 (selected_optimized_path == in handle_broadcast_mul()
[all …]
|
| H A D | op_div.cpp | 102 auto selected_optimized_path = select_optimized_path(a, b, out); in opt_div_out() local
103 if (selected_optimized_path == ElementwiseOptimizedPath::kTreatAs1d) { in opt_div_out()
122 } else if (selected_optimized_path != ElementwiseOptimizedPath::kNone) { in opt_div_out()
125 if (selected_optimized_path == in opt_div_out()
132 selected_optimized_path == in opt_div_out()
146 if (selected_optimized_path == in opt_div_out()
|
| H A D | op_sub.cpp | 141 auto selected_optimized_path = select_optimized_path(a, b, out); in opt_sub_out() local
142 if (selected_optimized_path == ElementwiseOptimizedPath::kTreatAs1d) { in opt_sub_out()
165 } else if (selected_optimized_path != ElementwiseOptimizedPath::kNone) { in opt_sub_out()
168 if (selected_optimized_path == in opt_sub_out()
175 selected_optimized_path == in opt_sub_out()
193 if (selected_optimized_path == in opt_sub_out()
|
| H A D | op_add.cpp | 118 auto selected_optimized_path = select_optimized_path(a, b, out); in opt_add_out() local
119 if (selected_optimized_path == ElementwiseOptimizedPath::kTreatAs1d) { in opt_add_out()
142 } else if (selected_optimized_path != ElementwiseOptimizedPath::kNone) { in opt_add_out()
145 if (selected_optimized_path == in opt_add_out()
152 selected_optimized_path == in opt_add_out()
|