Function test_broadcast_shape_compatibility

tests/test_graph.cpp:490–506

}

bool test_broadcast_shape_compatibility() {
    TestUtils::FP16TestFixture fixture("Broadcast Shape Compatibility");

    size_t a_id = fixture.create_input({2, 3});
    size_t b_id = fixture.create_input({2, 1});

    std::vector<__fp16> data_a = {1, 2, 3, 4, 5, 6};
    std::vector<__fp16> data_b = {10, 20};
    fixture.set_input_data(a_id, data_a);
    fixture.set_input_data(b_id, data_b);

    size_t result_id = fixture.graph().add(a_id, b_id);
    fixture.execute();

    std::vector<__fp16> expected = {11, 12, 13, 24, 25, 26};
    return fixture.verify_output(result_id, expected);
}

bool test_broadcast_scalar_tensor() {
    TestUtils::FP16TestFixture fixture("Broadcast Scalar Tensor");