Function make_bcast_shape

src/common.cpp:238–251

}

shape make_bcast_shape(const shape& input_shape, const std::vector<std::size_t>& bcast_lens)
{
    assert(not input_shape.dynamic());
    auto offset = bcast_lens.size() - input_shape.ndim();
    std::vector<size_t> bcast_strides(bcast_lens.size(), 0);
    for(std::ptrdiff_t i : reverse(range(input_shape.ndim())))
    {
        if(bcast_lens.at(i + offset) == input_shape.lens()[i])
        {
            bcast_strides.at(i + offset) = input_shape.strides()[i];
        }
    }
    return shape{input_shape.type(), bcast_lens, bcast_strides};
}

} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx