| 128 | } |
| 129 | |
| 130 | void DecomposeBiasTensor(const int32_t* biases, int bias_size, |
| 131 | std::vector<int32_t>* input_bias, |
| 132 | std::vector<int32_t>* cell_bias, |
| 133 | std::vector<int32_t>* forget_bias, |
| 134 | std::vector<int32_t>* output_bias) { |
| 135 | input_bias->resize(bias_size); |
| 136 | std::copy(biases, biases + bias_size, input_bias->begin()); |
| 137 | |
| 138 | cell_bias->resize(bias_size); |
| 139 | std::copy(biases + bias_size, biases + 2 * bias_size, cell_bias->begin()); |
| 140 | |
| 141 | forget_bias->resize(bias_size); |
| 142 | std::copy(biases + 2 * bias_size, biases + 3 * bias_size, |
| 143 | forget_bias->begin()); |
| 144 | |
| 145 | output_bias->resize(bias_size); |
| 146 | std::copy(biases + 3 * bias_size, biases + 4 * bias_size, |
| 147 | output_bias->begin()); |
| 148 | } |
| 149 | |
| 150 | } // namespace nnapi |
| 151 | } // namespace delegate |