Function create_test_node

internal/bin/test_eltwise_op.cpp:168–207

}

int create_test_node(graph_t graph, const char* node_name, const char* input_name, const char* input_name_1, int i_type)
{
    node_t test_node = create_graph_node(graph, node_name, "Eltwise");

    const int type = i_type;
    set_node_attr_int(test_node, "type", &type);
    const float power = 1;
    const float shift = 0;
    const float scale = 1;
    set_node_attr_float(test_node, "power", &power);
    set_node_attr_float(test_node, "shift", &shift);
    set_node_attr_float(test_node, "scale", &scale);

    tensor_t input_tensor = get_graph_tensor(graph, input_name);
    tensor_t input_tensor_1 = get_graph_tensor(graph, input_name_1);
    int data_type = get_tensor_data_type(input_tensor);

    if(input_tensor == nullptr)
    {
        std::cout << "ERRNO: " << get_tengine_errno() << "\n";
        return -1;
    }

    set_node_input_tensor(test_node, 0, input_tensor);
    set_node_input_tensor(test_node, 1, input_tensor_1);

    release_graph_tensor(input_tensor);
    release_graph_tensor(input_tensor_1);

    /* output */
    tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);

    set_node_output_tensor(test_node, 0, output_tensor, TENSOR_TYPE_VAR);

    release_graph_tensor(output_tensor);

    release_graph_node(test_node);

    return 0;
}

graph_t create_test_graph(const char* test_node_name, int c, int h, int w, int layout, int data_type, int i_type,
                          int case_num)