Method prepare

src/runtime/NEON/functions/NEQLSTMLayer.cpp:1464–1589

}

void NEQLSTMLayer::prepare()
{
    if (!_is_prepared)
    {
        if (_convert_input_to_forget_weights_to_qsymm8)
        {
            _input_to_forget_weights_f32.allocator()->allocate();
            _input_to_forget_weights_symm8.allocator()->allocate();
            _dequantize_input_to_forget_weights.run();
            _quantize_input_to_forget_weights.run();
        }

        // Pre-transpose weights to be used in GEMM.
        _input_to_forget_weights_transposed.allocator()->allocate();
        _input_to_cell_weights_transposed.allocator()->allocate();
        _input_to_output_weights_transposed.allocator()->allocate();
        _recurrent_to_forget_weights_transposed.allocator()->allocate();
        _recurrent_to_cell_weights_transposed.allocator()->allocate();
        _recurrent_to_output_weights_transposed.allocator()->allocate();
        _transpose_input_to_forget_weights.run();
        _transpose_input_to_cell_weights.run();
        _transpose_input_to_output_weights.run();
        _transpose_recurrent_to_forget_weights.run();
        _transpose_recurrent_to_cell_weights.run();
        _transpose_recurrent_to_output_weights.run();

        // Precompute effective biases
        if (_has_cifg)
        {
            std::fill_n(reinterpret_cast<int16_t *>(_ones.buffer()),
                        _ones.info()->total_size() / _ones.info()->element_size(), 32767);
        }
        else
        {
            _input_to_input_eff_bias.allocator()->allocate();
            _recurrent_to_input_eff_bias.allocator()->allocate();

            ITensorPack packII = {{TensorType::ACL_SRC, _input_to_input_weights},
                                  {TensorType::ACL_DST, &_input_to_input_eff_bias}};
            NEScheduler::get().schedule_op(_input_to_input_reduction.get(), Window::DimY,
                                           _input_to_input_reduction->window(), packII);

            ITensorPack packRI = {{TensorType::ACL_SRC, _recurrent_to_input_weights},
                                  {TensorType::ACL_DST, &_recurrent_to_input_eff_bias}};
            NEScheduler::get().schedule_op(_recurrent_to_input_reduction.get(), Window::DimY,
                                           _recurrent_to_input_reduction->window(), packRI);

            _input_to_input_weights_transposed.allocator()->allocate();
            _recurrent_to_input_weights_transposed.allocator()->allocate();
            _transpose_input_to_input_weights.run();
            _transpose_recurrent_to_input_weights.run();
            _input_to_input_weights->mark_as_unused();
            _recurrent_to_input_weights->mark_as_unused();
        }
        _input_to_forget_eff_bias.allocator()->allocate();
        _recurrent_to_forget_eff_bias.allocator()->allocate();
        _input_to_cell_eff_bias.allocator()->allocate();
        _recurrent_to_cell_eff_bias.allocator()->allocate();