Searched refs:prev_activ_dims (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 61 const Dims<4>& prev_activ_dims, const uint8_t* weights_data_uint8, in quantizedLstmStep() argument
72 MatchingFlatSizeSkipDim(input_dims, 0, prev_activ_dims, prevCellState_dims, in quantizedLstmStep()
77 const int prev_activ_depth = ArraySize(prev_activ_dims, 0); in quantizedLstmStep()
86 MatchingArraySize(prevCellState_dims, 0, prev_activ_dims, 0, in quantizedLstmStep()
98 Dims<4> const* concat_input_arrays_dims[2] = {&input_dims, &prev_activ_dims}; in quantizedLstmStep()
|