Searched refs:n_fw_input (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 23 n_fw_input, argument
88 "{{{}, {}, {}}}".format(max_time, n_batch, n_fw_input))
92 "{{{}, {}}}".format(n_cell, n_fw_input))
95 "{{{}, {}}}".format(n_cell, n_fw_input))
97 "{{{}, {}}}".format(n_cell, n_fw_input))
100 "{{{}, {}}}".format(n_cell, n_fw_input))
456 n_fw_input = 2 variable
465 n_fw_input=n_fw_input,
504 n_fw_input = 3 variable
513 n_fw_input=n_fw_input,
[all …]
|
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 271 const uint32_t n_fw_input = SizeOfDimension(input_, 2); in Prepare() local
275 NN_CHECK_EQ(SizeOfDimension(fw_input_to_output_weights_, 1), n_fw_input); in Prepare()
297 fw_output_layer_norm_weights_, n_fw_input, n_fw_output, n_fw_cell, ¶ms_)) { in Prepare()
|