Function shuffleBlock

benchmark/exprModels/ShuffleNetExpr.cpp:37–44

}

static VARP shuffleBlock(VARP x, int inputChannel, int outputChannel,
                         int group, int stride, int narrowRatio, int number) {
    x = shuffleUnit(x, inputChannel, outputChannel, group, stride, narrowRatio);
    for (int i = 1; i < number; ++i) {
        x = shuffleUnit(x, outputChannel, outputChannel, group, 1, narrowRatio);
    }
    return x;
}

VARP shuffleNetExpr(int group, int numClass) {
    std::map<int, std::vector<int>> groupMap = {