Function test_softmax

tests/test_graph.cpp:337–349

}

bool test_softmax() {
    TestUtils::FP16TestFixture fixture("Softmax");

    size_t input_a = fixture.create_input({2, 3});
    size_t softmax_result = fixture.graph().softmax(input_a, -1);

    std::vector<__fp16> input_data = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f};
    fixture.set_input_data(input_a, input_data);
    fixture.execute();

    std::vector<__fp16> expected = {0.09003f, 0.24473f, 0.66524f, 0.09003f, 0.24473f, 0.66524f};
    return fixture.verify_output(softmax_result, expected, 0.01f);
}

bool test_attention() {
    TestUtils::FP16TestFixture fixture("Attention");