Function set_input_data

internal/bin/test_concat_op.cpp:207–284

}

void set_input_data(graph_t graph, int input_counts)
{
    for(int ii = 0; ii < input_counts; ++ii)
    {
        tensor_t input_tensor = get_graph_input_tensor(graph, ii, 0);

        int data_type = get_tensor_data_type(input_tensor);
        int buf_size = get_tensor_buffer_size(input_tensor);
        void* i_buf = get_mem(buf_size);

        int dims[4];

        int num = get_tensor_shape(input_tensor, dims, 4);
        int elem_num = 1;
        while(num != 0)
        {
            elem_num *= dims[num - 1];
            num--;
        }

        // std::cout << "set input data counts : " << elem_num << "\n";

        for(int i = 0; i < elem_num; i++)
        {
            int iVal = i + 1;
            if(i == elem_num - 1)
            {
                iVal = 127;
            }

            if(data_type == TENGINE_DT_FP32)
            {
                float* f = ( float* )i_buf;
                f[i] = iVal;
                // std::cout << f[i] << "   ";
            }
            else if(data_type == TENGINE_DT_FP16)
            {
                __fp16* f16 = ( __fp16* )i_buf;

#ifdef __ARM_ARCH
                f16[i] = iVal;
#else
                f16[i] = fp32_to_fp16(iVal);
#endif
                // std::cout << iVal << "   ";
            }
            else if(data_type == TENGINE_DT_UINT8)
            {
                uint8_t* i8 = ( uint8_t* )i_buf;
                i8[i] = iVal;
                // std::cout << (int)i8[i] << "   ";
            }
            else
            {
                int8_t* i8 = ( int8_t* )i_buf;
                i8[i] = iVal;
                // std::cout << (int)i8[i] << "   ";