Method prepare

src/runtime/CL/functions/CLQLSTMLayer.cpp:1343–1452

}

void CLQLSTMLayer::prepare()
{
    if (!_is_prepared)
    {
        // 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)
        {
            _ones.map(true);
            std::fill_n(reinterpret_cast<int16_t *>(_ones.buffer()),
                        _ones.info()->total_size() / _ones.info()->element_size(), 32767);
            _ones.unmap();
        }
        else
        {
            _input_to_input_eff_bias.allocator()->allocate();
            _recurrent_to_input_eff_bias.allocator()->allocate();

            ITensorPack input_to_input_red_pack = {{ACL_SRC, _input_to_input_weights},
                                                   {ACL_DST, &_input_to_input_eff_bias}};
            CLScheduler::get().enqueue_op(*_input_to_input_reduction, input_to_input_red_pack, false);

            ITensorPack rec_to_input_red_pack = {{ACL_SRC, _recurrent_to_input_weights},
                                                 {ACL_DST, &_recurrent_to_input_eff_bias}};
            CLScheduler::get().enqueue_op(*_recurrent_to_input_reduction, rec_to_input_red_pack, false);

            _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();
        _input_to_output_eff_bias.allocator()->allocate();
        _recurrent_to_output_eff_bias.allocator()->allocate();

        ITensorPack input_to_forget_red_pack = {{ACL_SRC, _input_to_forget_weights},
                                                {ACL_DST, &_input_to_forget_eff_bias}};
        CLScheduler::get().enqueue_op(*_input_to_forget_reduction, input_to_forget_red_pack, false);

        ITensorPack rec_to_forget_red_pack = {{ACL_SRC, _recurrent_to_forget_weights},