Function PrepareTensors

executor/lib/custom_kernel.cpp:34–102

}

bool PrepareTensors(Node* node, struct custom_kernel_tensor**& k_inputs, int input_num,
                    struct custom_kernel_tensor**& k_outputs, int output_num)
{
    k_inputs = nullptr;
    k_outputs = nullptr;

    /*prepare inputs and outputs */
    if(input_num)
    {
        k_inputs = ( struct custom_kernel_tensor** )malloc(sizeof(void*) * input_num);

        if(k_inputs == nullptr)
        {
            set_tengine_errno(ENOMEM);
            return false;
        }

        memset(k_inputs, 0x0, sizeof(void*) * input_num);
    }

    if(output_num)
    {
        k_outputs = ( struct custom_kernel_tensor** )malloc(sizeof(void*) * output_num);

        if(k_outputs == nullptr)
        {
            set_tengine_errno(ENOMEM);
            return false;
        }

        memset(k_outputs, 0x0, sizeof(void*) * output_num);
    }

    for(int i = 0; i < input_num; i++)
    {
        struct custom_kernel_tensor* t = ( struct custom_kernel_tensor* )malloc(sizeof(struct custom_kernel_tensor));

        if(t == nullptr)
        {
            set_tengine_errno(ENOMEM);
            return false;
        }

        Tensor* tensor = node->GetInputTensor(i);

        PrepareOneTensor(node, tensor, t);

        k_inputs[i] = t;
    }

    for(int i = 0; i < output_num; i++)
    {
        struct custom_kernel_tensor* t = ( struct custom_kernel_tensor* )malloc(sizeof(struct custom_kernel_tensor));

        if(t == nullptr)
        {
            set_tengine_errno(ENOMEM);
            return false;