Searched refs:weights_dims (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 62 const Dims<4>& weights_dims, const int32_t* bias_data_int32, in quantizedLstmStep() argument
74 TFLITE_CHECK_EQ(ArraySize(weights_dims, 2), 1); in quantizedLstmStep()
75 TFLITE_CHECK_EQ(ArraySize(weights_dims, 3), 1); in quantizedLstmStep()
79 TFLITE_CHECK_EQ(ArraySize(weights_dims, 0), total_input_depth); in quantizedLstmStep()
83 MatchingArraySize(weights_dims, 1, bias_dims, 0); in quantizedLstmStep()
91 MatchingArraySize(weights_dims, 1, activ_temp_dims, 0); in quantizedLstmStep()
92 const int fc_accum_depth = ArraySize(weights_dims, 0); in quantizedLstmStep()
|