Searched refs:DecomposeBiasTensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | quant_lstm_sup_test.cc | 244 using tflite::delegate::nnapi::DecomposeBiasTensor;
246 TEST(DecomposeBiasTensor, ExtractInputBias) { in TEST() argument
263 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
269 TEST(DecomposeBiasTensor, ExtractCellBias) { in TEST() argument
286 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
292 TEST(DecomposeBiasTensor, ExtractForgetBias) { in TEST() argument
309 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
315 TEST(DecomposeBiasTensor, ExtractOutputBias) { in TEST() argument
332 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
|
| D | quant_lstm_sup.h | 48 void DecomposeBiasTensor(const int32_t* biases, int bias_size,
|
| D | quant_lstm_sup.cc | 130 void DecomposeBiasTensor(const int32_t* biases, int bias_size, in DecomposeBiasTensor() function
|
| D | nnapi_delegate.cc | 3302 delegate::nnapi::DecomposeBiasTensor(biases_tensor.data.i32, bias_size, in Map()
|