| 244 | using tflite::delegate::nnapi::DecomposeBiasTensor; |
| 245 | |
| 246 | TEST(DecomposeBiasTensor, ExtractInputBias) { |
| 247 | // clang-format off |
| 248 | std::vector<int32_t> biases |
| 249 | // inputGateBias |
| 250 | {-7876, 13488, -726, 32839, |
| 251 | // cellGateBias |
| 252 | 39481, 48624, 48976, -21419, |
| 253 | // forgetGateBias |
| 254 | 9206, -46884, -11693, -38724, |
| 255 | // outputGateBias |
| 256 | -58999, -17050, -41852, -40538}; |
| 257 | // clang-format on |
| 258 | |
| 259 | std::vector<int32_t> input_bias; |
| 260 | std::vector<int32_t> cell_bias; |
| 261 | std::vector<int32_t> forget_bias; |
| 262 | std::vector<int32_t> output_bias; |
| 263 | DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, |
| 264 | &output_bias); |
| 265 | |
| 266 | EXPECT_THAT(input_bias, ElementsAreArray({-7876, 13488, -726, 32839})); |
| 267 | } |
| 268 | |
| 269 | TEST(DecomposeBiasTensor, ExtractCellBias) { |
| 270 | // clang-format off |
nothing calls this directly
no test coverage detected