Home
last modified time | relevance | path

Searched refs:total_input_depth (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQuantizedLSTM.cpp74 const int total_input_depth = prev_activ_depth + input_depth; in quantizedLstmStep() local
75 TFLITE_CHECK_EQ(ArraySize(weights_dims, 0), total_input_depth); in quantizedLstmStep()