Searched refs:prev_activ_dims (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 60 const Dims<4>& prev_activ_dims, const uint8_t* weights_data_uint8, in quantizedLstmStep() argument
71 MatchingFlatSizeSkipDim(input_dims, 0, prev_activ_dims, prevCellState_dims, in quantizedLstmStep()
76 const int prev_activ_depth = ArraySize(prev_activ_dims, 0); in quantizedLstmStep()
85 MatchingArraySize(prevCellState_dims, 0, prev_activ_dims, 0, in quantizedLstmStep()
97 Dims<4> const* concat_input_arrays_dims[2] = {&input_dims, &prev_activ_dims}; in quantizedLstmStep()
|