| 39 | } |
| 40 | |
| 41 | static migraphx::instruction_ref |
| 42 | create_im2col(migraphx::instruction_ref& l_img, size_t channels, migraphx::module& m) |
| 43 | { |
| 44 | size_t f[2] = {1, 1}; |
| 45 | std::vector<int32_t> weights(channels * f[0] * f[1]); |
| 46 | migraphx::shape s_weights{migraphx::shape::int32_type, {1, channels, f[0], f[1]}}; |
| 47 | auto l_weights = m.add_literal(migraphx::literal{s_weights, weights}); |
| 48 | return m.add_instruction( |
| 49 | migraphx::make_op("im2col", {{"padding", {0, 0, 1, 1}}}), l_img, l_weights); |
| 50 | } |
| 51 | |
| 52 | static migraphx::instruction_ref |
| 53 | create_conv(migraphx::instruction_ref& l_img, |
no test coverage detected