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

Function create_im2col

test/eliminate_pad_test.cpp:42–50  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

40}
41
42static migraphx::instruction_ref
43create_im2col(migraphx::instruction_ref& l_img, size_t channels, migraphx::module& m)
44{
45 size_t f[2] = {1, 1};
46 std::vector<int32_t> weights(channels * f[0] * f[1]);
47 migraphx::shape s_weights{migraphx::shape::int32_type, {1, channels, f[0], f[1]}};
48 auto l_weights = m.add_literal(migraphx::literal{s_weights, weights});
49 return m.add_instruction(migraphx::make_op("im2col"), 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