| 881 | } |
| 882 | |
| 883 | void NELSTMLayer::prepare() |
| 884 | { |
| 885 | if (!_is_prepared) |
| 886 | { |
| 887 | _concat_weights_forget_gate.run(); |
| 888 | if (!_run_cifg_opt) |
| 889 | { |
| 890 | _concat_weights_input_gate.run(); |
| 891 | } |
| 892 | _concat_weights_output.run(); |
| 893 | _is_prepared = true; |
| 894 | } |
| 895 | } |
| 896 | } // namespace arm_compute |