| 331 | } |
| 332 | |
| 333 | std::unordered_map<instruction_ref, instruction_ref> |
| 334 | map_submod_params_to_inputs(module_ref submod, |
| 335 | const std::vector<instruction_ref>& group_inputs) const |
| 336 | { |
| 337 | auto map_param_to_main = submod->get_ins_param_map(group_inputs, true); |
| 338 | // verify the mapping is correct |
| 339 | auto expected_inputs = submod->get_inputs(map_param_to_main); |
| 340 | assert(expected_inputs == group_inputs and "Mapped inputs don't match group inputs"); |
| 341 | return map_param_to_main; |
| 342 | } |
| 343 | |
| 344 | void rebuild_attention_submodule( |
| 345 | module& target_mod, |
nothing calls this directly
no test coverage detected