MCPcopy Create free account
hub / github.com/ROCm/AMDMIGraphX / create_im2col

Function create_im2col

test/insert_pad_test.cpp:41–50  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

39}
40
41static migraphx::instruction_ref
42create_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
52static migraphx::instruction_ref
53create_conv(migraphx::instruction_ref& l_img,

Callers 1

TEST_CASEFunction · 0.70

Calls 3

make_opFunction · 0.50
add_literalMethod · 0.45
add_instructionMethod · 0.45

Tested by

no test coverage detected