Function get_group_conv_args

dnn/test/rocm/convolution.cpp:49–91

}

std::vector<TestArg> get_group_conv_args() {
    std::vector<TestArg> args;
    for (size_t batch_size : {2}) {
        for (size_t ih : {23}) {
            for (size_t iw : {ih + 1}) {
                for (size_t icpg : {2, 4, 8}) {
                    for (size_t ocpg : {4, 8}) {
                        for (size_t fh : {3, 5, 7}) {
                            for (size_t fw : {fh, fh + 1}) {
                                for (size_t ph : {0_z, size_t{fw / 2}}) {
                                    for (size_t sh : {1, 2}) {
                                        for (size_t dh : {1, 2}) {
                                            param::Convolution param;
                                            size_t groups = 2;
                                            param.sparse =
                                                    param::Convolution::Sparse::GROUP;

                                            param.mode = param::Convolution::Mode::
                                                    CROSS_CORRELATION;
                                            param.stride_h = param.stride_w = sh;
                                            param.pad_h = param.pad_w = ph;
                                            param.dilate_h = param.dilate_w = dh;
                                            args.emplace_back(
                                                    param,
                                                    TensorShape{
                                                            batch_size, icpg * groups,
                                                            ih, iw},
                                                    TensorShape{
                                                            groups, ocpg, icpg, fh,
                                                            fw});
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
    }

    return args;
}
}  // namespace convolution

TEST_F(ROCM, CONV_GROUP) {