Method configure_mm

src/runtime/NEON/functions/NEQLSTMLayer.cpp:203–229

}

void NEQLSTMLayer::configure_mm(NEGEMMLowpMatrixMultiplyCore &mm,
                                NEGEMMLowpOutputStage        &outstage,
                                GEMMLowpOutputStageInfo      &gemmlowp_info,
                                const ITensor                *mm_input,
                                const ITensor                *mm_weights,
                                const ITensor                *bias,
                                Tensor                       *mm_res,
                                Tensor                       *outstage_res,
                                float                         gemmlowp_scale,
                                const TensorInfo             &mm_res_info,
                                const TensorInfo             &outstage_tensor_info)
{
    _memory_group.manage(mm_res);
    _memory_group.manage(outstage_res);

    mm_res->allocator()->init(mm_res_info);
    outstage_res->allocator()->init(outstage_tensor_info);

    // Configure matrix-multiplication
    mm.configure(mm_input, mm_weights, nullptr, mm_res);

    // Configure output stage
    quantization::calculate_quantized_multiplier(gemmlowp_scale, &gemmlowp_info.gemmlowp_multiplier,
                                                 &gemmlowp_info.gemmlowp_shift);
    outstage.configure(mm_res, bias, outstage_res, gemmlowp_info);
    mm_res->allocator()->allocate();
}

void NEQLSTMLayer::configure(const ITensor             *input,
                             const ITensor             *input_to_forget_weights,