Function makeComplexGraph

test/expr/StaticModuleOutputReuseTest.cpp:11–44

using namespace MNN::Express;

static VARPS makeComplexGraph(VARP x) {
    // Input: NCHW float, shape {1, 4, 32, 32}
    // Graph intent:
    // - Introduce multiple ops (convert/conv/pool/concat/transpose) to increase
    //   intermediate allocations.
    // - Keep an early large tensor as an output (Aux) to make later output tensor
    //   more likely to be allocated with non-zero offset on Metal.

    auto x4 = _Convert(x, NC4HW4);

    auto c0 = _Conv(0.01f, 0.0f, x4, {4, 8}, {3, 3}, SAME, {1, 1}, {1, 1}, 1);
    c0      = _Relu(c0);

    auto maxP = _MaxPool(c0, {2, 2}, {2, 2}, VALID);
    auto aveP = _AvePool(c0, {2, 2}, {2, 2}, VALID);

    auto aux = _Concat({maxP, aveP}, 1);
    aux->setName("Aux");

    auto c1 = _Conv(0.02f, 0.01f, aux, {16, 4}, {1, 1}, SAME, {1, 1}, {1, 1}, 1);
    c1      = _Relu6(c1);

    auto y = _Convert(c1, NCHW);
    y      = _Transpose(y, {0, 2, 3, 1});
    y      = _Transpose(y, {0, 3, 1, 2});
    y      = y + _Scalar<float>(1.0f);
    y = _ReduceSum(y, {2}, true);
    y->setName("Output");

    auto s = _Shape(y);
    s->setName("Shape");

    return {aux, y, s};
}

static VARP makeInput(float base) {
    auto x   = _Input({1, 4, 32, 32}, NCHW, halide_type_of<float>());