Function test_gemm

src/driver/models.cpp:33–41

inline namespace MIGRAPHX_INLINE_NS {

migraphx::program test_gemm()
{
    migraphx::program p;
    auto* mm = p.get_main_module();
    auto a   = mm->add_parameter("a", migraphx::shape{migraphx::shape::float_type, {4, 5}});
    auto b   = mm->add_parameter("b", migraphx::shape{migraphx::shape::float_type, {5, 3}});
    mm->add_instruction(migraphx::make_op("dot"), a, b);
    return p;
}

} // namespace MIGRAPHX_INLINE_NS
} // namespace driver