Searched refs:n_bw_input (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 24 n_bw_input, argument
137 "{{{}, {}}}".format(n_cell, n_bw_input))
140 "{{{}, {}}}".format(n_cell, n_bw_input))
142 "{{{}, {}}}".format(n_cell, n_bw_input))
145 "{{{}, {}}}".format(n_cell, n_bw_input))
191 "{{{}, {}, {}}}".format(max_time, n_batch, n_bw_input))
457 n_bw_input = 2 variable
466 n_bw_input=n_bw_input,
505 n_bw_input = 2 variable
514 n_bw_input=n_bw_input,
|
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 394 const uint32_t n_bw_input = SizeOfDimension(bw_input, 2); in Prepare() local
405 bw_output_layer_norm_weights_, n_bw_input, n_bw_output, n_bw_cell, ¶ms_)) { in Prepare()
|