Function main

internal/bin/test_softmax.cpp:187–296

}

int main(int argc, char* argv[])
{
    int c = 1, h = 224, w = 224;
    const char* test_node_name = "Softmax";
    int data_type = TENGINE_DT_FP32;
    int layout = TENGINE_LAYOUT_NCHW;
    float negative_slope=0.0;

    init_tengine();

    graph_t graph = create_test_graph(test_node_name, c, h, w, layout, data_type, negative_slope);
    graph_t graph1 = create_test_graph(test_node_name, c, h, w, layout, data_type, negative_slope);

    if(graph == nullptr || graph1 == nullptr)
        return -1;

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

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

    for(unsigned int i = 0; i < buf_size/sizeof(float) ; i++)
    {
        i_buf[i] = (rand()%10)*0.01;
        i_buf1[i] = i_buf[i];
    }

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

    // prerun graph
    if(prerun_graph(graph) < 0)
    {
        std::cerr << "prerun_graph failed: ERRNO: " << get_tengine_errno() << "\n";
        return 1;
    }
    // prerun graph1
    setenv("OPS_REGISTRY","reference",1);
    setenv("OP_NAME","Softmax",1);
    if(prerun_graph(graph1) < 0 )
    {
        std::cerr << "prerun_graph1 failed: ERRNO: " << get_tengine_errno() << "\n";
        return 1;
    }
    unsetenv("OPS_REGISTRY");
    unsetenv("OP_NAME");

    // get device
    node_t test_node = get_graph_node(graph, test_node_name);
    node_t test_node1 = get_graph_node(graph1, test_node_name);
    // const char* dev = get_node_device(test_node);

    // std::cout << "node running on dev: " << dev << "\n";