Searched refs:weights_dims (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 61 const Dims<4>& weights_dims, const int32_t* bias_data_int32, in quantizedLstmStep() argument
73 TFLITE_CHECK_EQ(ArraySize(weights_dims, 2), 1); in quantizedLstmStep()
74 TFLITE_CHECK_EQ(ArraySize(weights_dims, 3), 1); in quantizedLstmStep()
78 TFLITE_CHECK_EQ(ArraySize(weights_dims, 0), total_input_depth); in quantizedLstmStep()
82 MatchingArraySize(weights_dims, 1, bias_dims, 0); in quantizedLstmStep()
90 MatchingArraySize(weights_dims, 1, activ_temp_dims, 0); in quantizedLstmStep()
91 const int fc_accum_depth = ArraySize(weights_dims, 0); in quantizedLstmStep()
|