Searched refs:bias_dims (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 63 const Dims<4>& bias_dims, const int16_t* prevCellState_data_int16, in quantizedLstmStep() argument
80 TFLITE_CHECK_EQ(MatchingArraySize(bias_dims, 1, bias_dims, 2, bias_dims, 3), in quantizedLstmStep()
83 MatchingArraySize(weights_dims, 1, bias_dims, 0); in quantizedLstmStep()
|