| 855 | } |
| 856 | |
| 857 | core::TensorValue graph_mapping_decoder( |
| 858 | core::ModuleBuildContext & ctx, |
| 859 | const core::TensorValue & input, |
| 860 | const std::unordered_map<std::string, Param> & params) { |
| 861 | auto x = graph_dense_block(ctx, input, params, "mask_decoder.dense_block"); |
| 862 | x = graph_subpixel_conv(ctx, x, params, "mask_decoder.mask_conv.0.conv1.weight", "mask_decoder.mask_conv.0.conv1.bias"); |
| 863 | x = graph_instance_norm_prelu( |
| 864 | ctx, |
| 865 | x, |
| 866 | params, |
| 867 | "mask_decoder.mask_conv.1.weight", |
| 868 | "mask_decoder.mask_conv.1.bias", |
| 869 | "mask_decoder.mask_conv.2.weight"); |
| 870 | x = graph_conv2d(ctx, x, params, "mask_decoder.mask_conv.3.weight", "mask_decoder.mask_conv.3.bias", 0, 0, 1, 1, 1, 1); |
| 871 | return modules::ReluModule().build(ctx, x); |
| 872 | } |
| 873 | |
| 874 | std::pair<core::TensorValue, core::TensorValue> graph_phase_decoder_real_imag( |
| 875 | core::ModuleBuildContext & ctx, |
no test coverage detected