Searched refs:prevCellState_dims (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 63 const Dims<4>& prevCellState_dims, int16_t* output_state_data_int16, in quantizedLstmStep() argument
71 MatchingFlatSizeSkipDim(input_dims, 0, prev_activ_dims, prevCellState_dims, in quantizedLstmStep()
85 MatchingArraySize(prevCellState_dims, 0, prev_activ_dims, 0, in quantizedLstmStep()
|