Home
last modified time | relevance | path

Searched defs:n_batch (Results 1 – 23 of 23) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTMTest.cpp76 LSTMOpModel(uint32_t n_batch, uint32_t n_input, in LSTMOpModel()
274 const int n_batch = 1; in TEST() local
385 const int n_batch = 1; in TEST() local
490 const int n_batch = 2; in TEST() local
DLSTM.cpp246 const uint32_t n_batch = SizeOfDimension(input, 0); in Prepare() local
303 const uint32_t n_batch = input_->shape().dimensions[0]; in Eval() local
/frameworks/ml/nn/runtime/test/specs/V1_0/
Dlstm2_state.mod.py21 n_batch = 1 variable
Dlstm3.mod.py21 n_batch = 2 variable
Dlstm3_state.mod.py21 n_batch = 2 variable
Dlstm3_state2.mod.py21 n_batch = 2 variable
Dlstm2.mod.py21 n_batch = 1 variable
Dlstm3_state3.mod.py21 n_batch = 2 variable
Dlstm2_state2.mod.py21 n_batch = 1 variable
Dlstm.mod.py21 n_batch = 1 variable
Dlstm_state2.mod.py21 n_batch = 1 variable
Dlstm_state.mod.py21 n_batch = 1 variable
/frameworks/ml/nn/runtime/test/specs/V1_1/
Dlstm2_state_relaxed.mod.py21 n_batch = 1 variable
Dlstm3_relaxed.mod.py21 n_batch = 2 variable
Dlstm2_state2_relaxed.mod.py21 n_batch = 1 variable
Dlstm2_relaxed.mod.py21 n_batch = 1 variable
Dlstm3_state3_relaxed.mod.py21 n_batch = 2 variable
Dlstm3_state_relaxed.mod.py21 n_batch = 2 variable
Dlstm3_state2_relaxed.mod.py21 n_batch = 2 variable
Dlstm_state2_relaxed.mod.py21 n_batch = 1 variable
Dlstm_relaxed.mod.py21 n_batch = 1 variable
Dlstm_state_relaxed.mod.py21 n_batch = 1 variable
/frameworks/ml/nn/tools/test_generator/tests/P_lstm/
Dlstm.mod.py21 n_batch = 1 variable