Function bottleNeck

benchmark/exprModels/MobileNetExpr.cpp:84–95

}

static VARP bottleNeck(VARP x, INTS channels, int stride, int expansionRatio) {
    int inputChannel = channels[0], outputChannel = channels[1];
    int expansionChannel = inputChannel * expansionRatio, group = expansionChannel;
    auto y = _Conv(0.0f, 0.0f, x, {inputChannel, expansionChannel}, {1, 1}, VALID, {1, 1}, {1, 1}, 1);
    y = _Conv(0.0f, 0.0f, y, {expansionChannel, expansionChannel}, {3, 3}, SAME, {stride, stride}, {1, 1}, group);
    y = _Conv(0.0f, 0.0f, y, {expansionChannel, outputChannel}, {1, 1}, VALID, {1, 1}, {1, 1}, 1);
    if (inputChannel != outputChannel || stride != 1) {
        x = _Conv(0.0f, 0.0f, x, {inputChannel, outputChannel}, {1, 1}, SAME, {stride, stride}, {1, 1}, 1);
    }
    y = _Add(x, y);
    return y;
}

static VARP bottleNeckBlock(VARP x, INTS channels, int stride, int expansionRatio, int number) {
    x = bottleNeck(x, {channels[0], channels[1]}, stride, expansionRatio);