Function compute_broadcasted_dyn_dims

src/common.cpp:54–90

    return out_lens;
}
std::vector<shape::dynamic_dimension>
compute_broadcasted_dyn_dims(std::vector<shape::dynamic_dimension> dds0,
                             std::vector<shape::dynamic_dimension> dds1)
{
    if(dds0.size() > dds1.size())
    {
        std::swap(dds0, dds1);
    }
    auto offset = dds1.size() - dds0.size();
    std::vector<shape::dynamic_dimension> out_dims(dds1);
    std::transform(dds0.cbegin(),
                   dds0.cend(),
                   dds1.cbegin() + offset,
                   out_dims.begin() + offset,
                   [&](auto a, auto b) {
                       if(a == b or b == 1)
                       {
                           return a;
                       }
                       else if(a == 1)
                       {
                           return b;
                       }
                       else
                       {
                           auto intersect = a.intersection(b);
                           if(intersect.has_value())
                           {
                               return intersect.value();
                           }
                           MIGRAPHX_THROW("COMPUTE_BROADCASTED_DYN_DIMS: dynamic shapes {" +
                                          migraphx::to_string_range(dds0) + "} and {" +
                                          migraphx::to_string_range(dds1) + "} mismatch!");
                       }
                   });
    return out_dims;
}

std::vector<shape::dynamic_dimension> compute_broadcasted_dyn_dims(shape s0, shape s1)
{