Searched refs:bias_dims (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 62 const Dims<4>& bias_dims, const int16_t* prevCellState_data_int16, in quantizedLstmStep() argument
79 TFLITE_CHECK_EQ(MatchingArraySize(bias_dims, 1, bias_dims, 2, bias_dims, 3), in quantizedLstmStep()
82 MatchingArraySize(weights_dims, 1, bias_dims, 0); in quantizedLstmStep()
|