Function test_cat

tests/test_graph.cpp:60–90

}

bool test_cat() {
    TestUtils::FP16TestFixture fixture("Cat (multiple input tensors)");

    size_t input_a = fixture.create_input({2, 3});
    size_t input_b = fixture.create_input({2, 5});
    size_t input_c = fixture.create_input({2, 2});

    size_t cat_result = fixture.graph().cat({input_a, input_b, input_c}, 1);

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

    std::vector<__fp16> data_b = {1, 2, 3, 4, 5,
                                  6, 7, 8, 9, 10};

    std::vector<__fp16> data_c = {-1, -2,
                                  -1, -2};

    std::vector<__fp16> expected = {
        1, 2, 3, 1, 2, 3, 4, 5, -1, -2,
        4, 5, 6, 6, 7, 8, 9, 10, -1, -2
    };

    fixture.set_input_data(input_a, data_a);
    fixture.set_input_data(input_b, data_b);
    fixture.set_input_data(input_c, data_c);

    fixture.execute();

    return fixture.verify_output(cat_result, expected);
}

bool test_view() {
    TestUtils::FP16TestFixture fixture("View");