Function main

internal/bin/test_generic.cpp:307–427

}

int main(int argc, char* argv[])
{
    int c, h, w;

    c = 3;
    h = 16;
    w = 16;

    init_tengine();

    graph_t graph = create_test_graph(c, h, w);

    if(graph == nullptr)
        return 1;

    /* using custom kernel to do cacluation */
    init_custom_kernel(&generic_custom_kernel);

    node_t my_node = get_graph_node(graph, "generic");

    fill_custom_kernel_param(my_node, &generic_custom_kernel);

    if(set_custom_kernel(my_node, "ANY_DEVICE", &generic_custom_kernel) < 0)
    {
        std::cerr << "set_custom_kernel failed: ERRNO: " << get_tengine_errno() << "\n";
        return 1;
    }

    release_graph_node(my_node);

    /* set input */
    tensor_t input_tensor = get_graph_input_tensor(graph, 0, 0);

    int buf_size = get_tensor_buffer_size(input_tensor);
    float* i_buf = ( float* )malloc(buf_size);

    for(unsigned int i = 0; i < buf_size / sizeof(float); i++)
        i_buf[i] = 1;

    set_tensor_buffer(input_tensor, i_buf, buf_size);
    release_graph_tensor(input_tensor);

    /* set weight */
    node_t conv_node = get_graph_node(graph, "conv");

    tensor_t weight_tensor = get_node_input_tensor(conv_node, 1);

    buf_size = get_tensor_buffer_size(weight_tensor);
    float* w_buf = ( float* )malloc(buf_size);

    for(unsigned int i = 0; i < buf_size / sizeof(float); i++)
        w_buf[i] = 1;

    set_tensor_buffer(weight_tensor, w_buf, buf_size);

    release_graph_tensor(weight_tensor);

    /* set bias */