Function run_shape_transforms

test/shape_transform_descriptor.cpp:82–98

}

static std::vector<int64_t> run_shape_transforms(const std::vector<std::size_t>& dims,
                                                 const std::vector<migraphx::operation>& ops)
{
    migraphx::shape s{migraphx::shape::int64_type, dims};
    std::vector<int64_t> data(s.elements());
    std::iota(data.begin(), data.end(), 0);

    migraphx::program p;
    auto* mm   = p.get_main_module();
    auto start = mm->add_literal(s, data);
    for(const auto& op : ops)
        start = mm->add_instruction(op, start);
    mm->add_return({start});

    auto result = p.eval({}).at(0);
    return result.to_vector<int64_t>();
}

static std::vector<int64_t> run_strided_view(const migraphx::shape& s, std::int64_t offset)
{