Function load_setsubtensor

runtime/src/vm/subtensor.c:207–240

#if ENABLE_INST_SETSUBTENSOR
static TinyNNStatus load_setsubtensor(
        flatbuffers_generic_t fbs_inst, Instruction* inst, VM* vm) {
    SetSubTensor* set_subtensor = &inst->workload.set_subtensor;
    ns(SetSubTensor_table_t) fbs_set_subtensor = (ns(SetSubTensor_table_t))(fbs_inst);
    inst->tag = TinyNN_INST_SETSUBTENSOR;
    flatbuffers_int32_vec_t fbs_inputs = ns(SetSubTensor_inputs(fbs_set_subtensor));
    flatbuffers_int8_vec_t fbs_input_types =
            ns(SetSubTensor_input_types(fbs_set_subtensor));
    set_subtensor->nr_input = flatbuffers_int32_vec_len(fbs_inputs);

    int total_input = set_subtensor->nr_input;
    set_subtensor->inputs = tinynn_malloc(total_input * sizeof(Tensor*));
    DeviceModel* model = get_active_device_model(vm);
    LOG_DEBUG("\t setsubtensor inputs tensor number:%d\n", set_subtensor->nr_input);
    //! parse the input
    parase_inputs(
            set_subtensor->inputs, total_input, model, vm->model, fbs_inputs,
            fbs_input_types);

    int32_t output_idx = ns(SetSubTensor_output(fbs_set_subtensor));
    set_subtensor->output = model->tensors + output_idx;

    ns(IndexDesc_vec_t) fbs_descs = ns(SetSubTensor_descs(fbs_set_subtensor));
    ns(IndexDesc_vec_t) fbs_flags = ns(SetSubTensor_flags(fbs_set_subtensor));
    parse_subtensor(&set_subtensor->descs, fbs_descs);
    parse_subtensor(&set_subtensor->flags, fbs_flags);
    TINYNN_ASSERT_MSG(
            ns(IndexDesc_vec_len(fbs_descs)) == ns(IndexDesc_vec_len(fbs_flags)),
            "The size of setsubtensor descs and flags is not equal.");
    set_subtensor->nr_descs = ns(IndexDesc_vec_len(fbs_descs));
    sort_descs(set_subtensor->descs, set_subtensor->nr_descs, set_subtensor->flags);

    return TinyNN_SUCCESS;
}
static TinyNNStatus execute_setsubtensor(Instruction* inst, VM* vm) {
    Tensor **inputs = inst->workload.set_subtensor.inputs,
           *output = inst->workload.set_subtensor.output;