Function CheckBind

executor/lib/custom_kernel.cpp:130–154

}

static bool CheckBind(Node* node, struct custom_kernel_ops* ops)
{
    /* prepare tensors */
    int input_num = node->GetInputNum();
    int output_num = node->GetOutputNum();

    struct custom_kernel_tensor** k_inputs;
    struct custom_kernel_tensor** k_outputs;

    if(!PrepareTensors(node, k_inputs, input_num, k_outputs, output_num))
    {
        ReleaseTensors(node, k_inputs, input_num, k_outputs, output_num);
        return false;
    }

    int ret = ops->bind(ops, ( const struct custom_kernel_tensor** )k_inputs, input_num,
                        ( const struct custom_kernel_tensor** )k_outputs, output_num);

    ReleaseTensors(node, k_inputs, input_num, k_outputs, output_num);

    if(ret < 0)
        return false;

    return true;
}

NodeOps* BindOps(Node* node, const std::string& dev_name)
{