| 506 | } |
| 507 | |
| 508 | auto generate(const program& p, |
| 509 | const target& t, |
| 510 | bool offload, |
| 511 | unsigned batch, |
| 512 | dims_map map_input_dims = {}) |
| 513 | { |
| 514 | parameter_map m; |
| 515 | auto param_shapes = p.get_parameter_shapes(); |
| 516 | std::unordered_map<std::string, shape> static_param_shapes; |
| 517 | for(auto&& param : param_shapes) |
| 518 | { |
| 519 | if(contains(map_input_dims, param.first)) |
| 520 | static_param_shapes[param.first] = {param.second.type(), |
| 521 | map_input_dims[param.first]}; |
| 522 | else |
| 523 | static_param_shapes[param.first] = param.second.to_static(batch); |
| 524 | } |
| 525 | |
| 526 | for(auto&& s : fill0) |
| 527 | m[s] = fill_argument(static_param_shapes.at(s), 0); |
| 528 | for(auto&& s : fill1) |
| 529 | m[s] = fill_argument(static_param_shapes.at(s), 1); |
| 530 | fill_param_map(m, static_param_shapes, t, offload); |
| 531 | auto load_arg_map = program_params::parse_load_args(load_args_info, t, offload); |
| 532 | for(auto&& arg : load_arg_map) |
| 533 | { |
| 534 | m[arg.first] = arg.second; |
| 535 | } |
| 536 | return m; |
| 537 | } |
| 538 | }; |
| 539 | |
| 540 | struct compiler_target |
no test coverage detected