| 50 | } |
| 51 | |
| 52 | static migraphx::instruction_ref |
| 53 | create_conv(migraphx::instruction_ref& l_img, |
| 54 | size_t channels, |
| 55 | migraphx::module& m, |
| 56 | migraphx::op::padding_mode_t padding_mode = migraphx::op::padding_mode_t::default_) |
| 57 | { |
| 58 | migraphx::shape s_weights{migraphx::shape::int32_type, {4, channels, 3, 3}}; |
| 59 | std::vector<int32_t> weights(4 * channels * 3 * 3); |
| 60 | auto l_weights = m.add_literal(migraphx::literal{s_weights, weights}); |
| 61 | return m.add_instruction( |
| 62 | migraphx::make_op("convolution", |
| 63 | {{"padding_mode", padding_mode}, {"padding", {0, 0, 1, 1}}}), |
| 64 | l_img, |
| 65 | l_weights); |
| 66 | } |
| 67 | |
| 68 | TEST_CASE(rewrite_pad) |
| 69 | { |
no test coverage detected