Function PrepareOneTensor

executor/lib/custom_kernel.cpp:12–32

namespace CustomKernelManager {

static void PrepareOneTensor(Node* node, Tensor* tensor, struct custom_kernel_tensor* t)
{
    const TShape& shape = tensor->GetShape();
    const std::vector<int>& dims = shape.GetDim();

    const ExecAttr* exec_attr = any_cast<const ExecAttr*>(node->GetAttr(ATTR_EXEC_ATTR));

    t->dim_num = dims.size();

    for(int i = 0; i < t->dim_num; i++)
    {
        t->dim[i] = dims[i];
    }

    t->data_type = tensor->GetDataType();
    t->element_num = shape.GetSize();
    t->element_size = DataType::GetTypeSize(t->data_type);
    t->layout_type = exec_attr->graph_layout;

    t->data = nullptr;
}

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