Function test_concat

tests/test_graph.cpp:43–58

}

bool test_concat() {
    TestUtils::FP16TestFixture fixture("Concat");

    size_t input_a = fixture.create_input({2, 3});
    size_t input_b = fixture.create_input({2, 5});
    size_t concat_result = fixture.graph().concat(input_a, input_b, 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> expected = {1, 2, 3, 1, 2, 3, 4, 5, 4, 5, 6, 6, 7, 8, 9, 10};
    fixture.set_input_data(input_a, data_a);
    fixture.set_input_data(input_b, data_b);
    fixture.execute();

    return fixture.verify_output(concat_result, expected);
}

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