Method compute_shape

src/targets/gpu/lrn.cpp:31–35

namespace gpu {
#if MIGRAPHX_USE_MIOPEN
shape miopen_lrn::compute_shape(const std::vector<shape>& inputs) const
{
    check_shapes{inputs, *this}.has(2).not_broadcasted();
    return inputs.at(1);
}

argument miopen_lrn::compute(context& ctx,
                             const shape& output_shape,