Function fireMoudle

benchmark/exprModels/SqueezeNetExpr.cpp:16–22

fire module in squeezeNet model

// fire module in squeezeNet model
static VARP fireMoudle(VARP x, int inputChannel, int squeeze_1x1,
                       int expand_1x1, int expand_3x3) {
    x = _Conv(0.0f, 0.0f, x, {inputChannel, squeeze_1x1}, {1, 1}, VALID, {1, 1}, {1, 1}, 1);
    auto y1 = _Conv(0.0f, 0.0f, x, {squeeze_1x1, expand_1x1}, {1, 1}, VALID, {1, 1}, {1, 1}, 1);
    auto y2 = _Conv(0.0f, 0.0f, x, {squeeze_1x1, expand_3x3}, {3, 3}, SAME, {1, 1}, {1, 1}, 1);
    return _Concat({y1, y2}, 1); // concat on channel axis (NCHW)
}

VARP squeezeNetExpr(int numClass) {
    auto x = _Input({1, 3, 224, 224}, NC4HW4);