Function set_input_data

internal/bin/test_mini_op.cpp:152–209

}

void* set_input_data(graph_t graph)
{
    tensor_t input_tensor = get_graph_input_tensor(graph, 0, 0);

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

    int dims[4];

    get_tensor_shape(input_tensor, dims, 4);

    int elem_num = dims[0] * dims[1];

    std::cout << "Dims[0]=" << dims[0] << " dims[1]=" << dims[1] << "elem_num=" << elem_num << "\n";
    tensor_t input_tensor1 = get_graph_input_tensor(graph, 1, 0);

    int data_type = get_tensor_data_type(input_tensor);
    std::cout << "data type = " << data_type;
    for(int i = 0; i < elem_num; i++)
    {
        if(data_type == TENGINE_DT_FP32)
        {
            float* f = ( float* )i_buf;
            f[i] = 0.63907;
            float* p = ( float* )i_buf1;
            p[i] = 0.98765;
        }
        else if(data_type == TENGINE_DT_FP16)
        {
            __fp16* f16 = ( __fp16* )i_buf;

#ifdef __ARM_ARCH
            f16[i] = -1.0;
#else
            f16[i] = fp32_to_fp16(-1.1);
#endif
        }
        else if(data_type == TENGINE_DT_INT8)
        {
            int8_t* int8 = ( int8_t* )i_buf;
            int8[i] = -32;
        }
        else
        {
            uint8_t* i8 = ( uint8_t* )i_buf;
            i8[i] = 32;
        }
    }

    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);

    return i_buf;
}