Method PrepareTensors

executor/lib/custom_kernel_ops.cpp:129–156

}

bool CustomKernelNodeOps::PrepareTensors(Node* node)
{
    /*prepare inputs and outputs */
    input_num_ = node->GetInputNum();
    output_num_ = node->GetOutputNum();

    if(!CustomKernelManager::PrepareTensors(node, k_inputs_, input_num_, k_outputs_, output_num_))
        return false;

    /* set mem in tensor */
    for(int i = 0; i < input_num_; i++)
    {
        Tensor* tensor = node->GetInputTensor(i);
        struct custom_kernel_tensor* t = k_inputs_[i];

        t->data = get_tensor_mem(tensor);
    }

    for(int i = 0; i < output_num_; i++)
    {
        Tensor* tensor = node->GetOutputTensor(i);
        struct custom_kernel_tensor* t = k_outputs_[i];

        t->data = get_tensor_mem(tensor);
    }

    return true;
}

void CustomKernelNodeOps::ReleaseTensors(Node* node)
{