Home
last modified time | relevance | path

Searched defs:n_cell (Results 1 – 25 of 56) sorted by relevance

123

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTMTest.cpp78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel()
276 const int n_cell = 4; in TEST() local
380 const int n_cell = 4; in TEST() local
476 const int n_cell = 20; in TEST() local
H A DLayerNormLSTMTest.cpp86 LayerNormLSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LayerNormLSTMOpModel()
289 const int n_cell = 4; in TEST() local
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
H A Dbidirectional_sequence_lstm.mod.py25 n_cell, argument
458 n_cell = 4 variable
506 n_cell = 4 variable
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
H A Dquantized_lstm.mod.py22 n_cell = 4 variable
109 n_cell = 4 variable
H A Dlayer_norm_lstm.mod.py24 n_cell = 4 variable
196 n_cell = 4 variable
H A Dlstm2_float16.mod.py24 n_cell = 4 variable
H A Dlstm2_state2_float16.mod.py24 n_cell = 4 variable
H A Dlstm2_state_float16.mod.py24 n_cell = 4 variable
H A Dlstm3_float16.mod.py24 n_cell = 20 variable
H A Dlstm3_state2_float16.mod.py24 n_cell = 20 variable
H A Dlstm3_state3_float16.mod.py24 n_cell = 20 variable
H A Dlstm3_state_float16.mod.py24 n_cell = 20 variable
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
H A Dlstm2.mod.py24 n_cell = 4 variable
H A Dlstm2_state.mod.py24 n_cell = 4 variable
H A Dlstm2_state2.mod.py24 n_cell = 4 variable
H A Dlstm3.mod.py24 n_cell = 20 variable
H A Dlstm3_state.mod.py24 n_cell = 20 variable
H A Dlstm3_state2.mod.py24 n_cell = 20 variable
H A Dlstm3_state3.mod.py24 n_cell = 20 variable
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
H A Dlstm2_relaxed.mod.py24 n_cell = 4 variable
H A Dlstm2_state2_relaxed.mod.py24 n_cell = 4 variable
H A Dlstm2_state_relaxed.mod.py24 n_cell = 4 variable
H A Dlstm3_relaxed.mod.py24 n_cell = 20 variable
H A Dlstm3_state2_relaxed.mod.py24 n_cell = 20 variable
H A Dlstm3_state3_relaxed.mod.py24 n_cell = 20 variable

123