Function test_flatten

tests/test_graph.cpp:107–120

}

bool test_flatten() {
    TestUtils::FP16TestFixture fixture("Flatten");

    size_t input = fixture.create_input({2, 3});
    size_t flatten_result = fixture.graph().flatten(input);

    std::vector<__fp16> data = {1, 2, 3, 4, 5, 6};
    std::vector<__fp16> expected = {1, 2, 3, 4, 5, 6};

    fixture.set_input_data(input, data);
    fixture.execute();

    return fixture.verify_output(flatten_result, expected);
}

bool test_basic_operations() {
    TestUtils::FP16TestFixture fixture("Basic Operations");