Function test_matrix_multiplication

tests/test_graph.cpp:187–202

}

bool test_matrix_multiplication() {
    TestUtils::FP16TestFixture fixture("Matrix Multiplication");

    size_t input_a = fixture.create_input({2, 3});
    size_t input_b = fixture.create_input({3, 2});
    size_t matmul_result = fixture.graph().matmul(input_a, input_b, false);

    std::vector<__fp16> data_a = {1, 2, 3, 4, 5, 6};
    std::vector<__fp16> data_b = {1, 2, 3, 4, 5, 6};
    fixture.set_input_data(input_a, data_a);
    fixture.set_input_data(input_b, data_b);
    fixture.execute();

    std::vector<__fp16> expected = {22, 28, 49, 64};
    return fixture.verify_output(matmul_result, expected);
}

bool test_transpose() {
    TestUtils::FP16TestFixture fixture("Transpose");